al

Binary Search - 二分查找

  • 能使用二分搜索的前提是数组已排序。
  • 递归和迭代的使用场景:能用迭代就用迭代,特别复杂时采用递归。

Question: lintcode - (14) Binary Search

Binary search is a famous question in algorithm.

For a given sorted array (ascending order) and a target number, find the first index of this number in O(log n) time complexity.

If the target number does not exist in the array, return -1.

Example

If the array is [1, 2, 3, 3, 4, 5, 10], for given target 3, return 2.

Challenge

If the count of numbers is bigger than MAXINT, can your code work properly?

题解:

对于已排序升序数组,使用二分查找可满足复杂度要求,注意数组中可能有重复值。

/**
 * 本代码fork自九章算法。没有版权欢迎转发。
 * http://www.jiuzhang.com//solutions/binary-search/
 */
class Solution {
    /**
     * @param nums: The integer array.
     * @param target: Target to find.
     * @return: The first position of target. Position starts from 0.
     */
    public int binarySearch(int[] nums, int target) {
        if (nums == null || nums.length == 0) {
            return -1;
        }

        int start = 0;
        int end = nums.length - 1;
        int mid;
        while (start + 1 < end) {
            mid = start + (end - start) / 2; // avoid overflow when (end + start)
            if (target < nums[mid]) {
                end = mid;
            } else if (target > nums[mid]) {
                start = mid;
            } else {
                end = mid;
            }
        }

        if (nums[start] == target) {
            return start;
        }
        if (nums[end] == target) {
            return end;
        }

        return -1;
    }
}

源码分析:

  1. 首先对输入做异常处理,数组为空或者长度为0
  2. 初始化 start, end, mid三个变量,注意mid的求值方法,可以防止两个整型值相加时溢出
  3. 使用迭代而不是递归进行二分查找
  4. while终止条件应为start + 1 < end而不是start <= endstart == end时可能出现死循环。即循环终止条件是相邻或相交元素时推出。
  5. 迭代终止时target应为start或者end中的一个——由上述循环终止条件有两个,具体谁先谁后视题目而定。
  6. 赋值语句end = mid有两个条件是相同的,可以选择写到一块。