二分查找算法

问题

如何确定一个元素在数组中的位置？（假设数组里面全是整数）

• 如果是无序数组，从第0个位置开始遍历搜索，平均时间复杂度：O(n)
  
• 如果是有序数组，可以使用二分搜索，最坏时间复杂度：O(log n)
  

思路

• 假设在[begin,end)（左闭右开，end对应的值不做比较）范围内搜索某个元素v，mid == (begin+end)/2
• 如果 v < m,去[begin,min)范围内二分搜索
• 如果 v > m,去[mid+1,end)范围内二分搜索
• 如果 v = m,直接返回mid
• 如果begin == end，表示已经不存在元素，说明没有找到，返回-1

案例

实现代码

public class 二分查找 {
    public static void main(String[] args) {
        int[] arr = {0,1,2,3,4,5,6,7,8,9,10};
        System.out.println(search(arr,12));
    }

    public static int search(int[] arr, int v) {
        if (arr==null || arr.length<0) return -1; // 如果数组为空则直接返回-1，表示搜索失败
        int begin = 0;  // begin初始值设为0
        int end = arr.length; // 左闭右开[begin,end)
        // 只有当begin<end 才满足查找条件，因为设计的是左闭右开，end是不存在的，如果begin=end，则表示数据已经不存在，没有找到
        while (begin < end) {
            int min = (begin+end)/2; // 获取中间值下标
            if (v < arr[min]) {
                end = min; // 如果v比中间值小，将end设置为min
            }else if (v > arr[min]) {
                begin = min + 1;    // 如果v比中间值大，将begin设置为min
            }else {
                return min;
            }
        }
        // 如果没有找到直接返回-1
        return -1;
    }
}

附加

找到数组中第一个大于某个值的下标？（插入排序优化时可以用到）
直接上代码：

/**
 * 找到数组中第一个大于某个值的下标（插入排序优化时可以用到）
 */
public static int firstBig(int[] arr, int v) {
    if (arr==null || arr.length<0) return -1; // 如果数组为空则直接返回-1，表示搜索失败
    int begin = 0;  // begin初始值设为0
    int end = arr.length; // 左闭右开[begin,end)
    while (begin < end) {
        int min = (begin+end) >> 1;
        if (v<arr[min]) {   // 因为我们要找的是第一个大于v的值，这里不能是v<=arr[min]，否则min之前的一段是找不到目标下标的
            end = min;
        }else if (v>=min) { // 因为我们要找的是第一个大于v的值，因此这个 '='必须与'>'关联到一块儿
            begin = min+1;
        }
    }
    return begin; // 等begin=end的时候，就是我们要找的下标
}