跳到主要内容

01、算法与数据结构 - 实战:二分法

普通二分查找

/**
 * 普通二分查找
 */
public class BinarySearch01 {
   
     

    public static int search(int[] arr, int num) {
   
     
        if (arr == null || arr.length == 0) return -1;
        int l = 0;
        int r = arr.length - 1;
        int mid;
        while (l <= r) {
   
     
            mid = l + ((r - l) >> 1);
            if (arr[mid] > num) r = mid - 1;
            else if (arr[mid] < num) l = mid + 1;
            else return mid;
        }
        return -1;
    }

}

在有序数组中查找大于等于某个数的最左侧位置

二分法不仅可以用于寻找某个指定的值,还可以寻找每个值出现的最左或最右位置,但是这种二分就不是寻找到指定的值就返回,而是继续二分,直到二分尽为止

/**
 * 二分查找大于等于某个数的最左侧位置
 * 如:数组 000111222333,num 1
 * 则返回的下标为3
 */
public class BinarySearch02 {
   
     

    public static int search(int[] arr, int num) {
   
     
        if (arr == null || arr.length == 0) return -1;
        int l = 0;
        int r = arr.length - 1;
        int index = -1;
        while (l <= r) {
   
     
            int mid = l + ((r - l) >> 1);
            if (arr[mid] >= num) {
   
     
                index = mid; // arr[mid]是目前已知的大于等于num的最左侧位置
                r = mid - 1; // 排查mid右边的数
            } else l = mid + 1; // mid以及mid左边的数都比num小,排查mid以及mid左边的数
        }
        return index;
    }

}

在有序数组中查找小于等于某个数的最右侧位置

/**
 * 二分查找小于等于某个数的最右侧位置
 * 如:数组 000111222333,num 5
 * 则返回的下标为3
 */
public class BinarySearch03 {
   
     

    public static int search(int[] arr, int num) {
   
     
        if (arr == null || arr.length == 0) return -1;
        int l = 0;
        int r = arr.length - 1;
        int index = -1;
        while (l <= r) {
   
     
            int mid = l + ((r - l) >> 1);
            if (arr[mid] > num) r = mid - 1; //mid以及mid右侧都是大于num的数,都排除
            else {
   
     
                index = mid; // 目前已知的小于等于num的最右侧位置
                l = mid + 1; // 排除左边的数
            }
        }
        return index;
    }

}

寻找局部最小

不一定是有序的数组才能二分,只要找到一种规律,可以一次排除掉一半的数据,就可以做二分
 
 

/**
 * 寻找局部最小
 * arr数组无序,并且相邻两个数不相等,从中找出一个局部最小的数,并返回下标
 * 局部最小:
 * 比如下标为i的数,如果小于小标i-1和i+1的数,则下标为i的数就是局部最小
 * 如果下标为0的数,小于下标为1的数,则下标为0的数就是局部最小
 * 如果下标为arr.length-1的数,小于arr.length-2的数,则arr.length-1的数就是局部最小
 */
public class BinarySearch04 {
   
     

    public static int search(int[] arr) {
   
     
        if (arr == null || arr.length == 0) return -1;
        if (arr.length == 1) return 0;

        //arr[0] < arr[1],所以arr[0]满足局部最小,返回0
        if (arr[0] < arr[1]) return 0;

        //arr[arr.length - 1] < arr[arr.length - 2],所以arr[arr.length - 1]满足局部最小,返回arr.length - 1
        if (arr[arr.length - 1] < arr[arr.length - 2]) return arr.length - 1;

        //此时l~r范围内,arr[l] > arr[l+1] 趋势递减,arr[arr.length-2] < arr[arr.length-1]趋势递增
        //中间必有最小值
        int l = 1;
        int r = arr.length - 2;
        while (l <= r) {
   
     
            int mid = l + ((r - l) >> 1);
            //找到局部最小,返回
            if (arr[mid] < arr[mid - 1] && arr[mid] < arr[mid + 1]) return mid;
            //如果arr[mid] > arr[mid - 1],则arr[mid-1] < arr[mid],趋势递增,l~mid中间必有最小值
            //r赋值为mid-1
            else if (arr[mid] > arr[mid - 1]) r = mid - 1;
            //arr[mid]小于arr[mid-1],只能排除左边的一部分,从右边的一部分寻找局部最小
            else l = mid + 1;
        }
        return -1;
    }

}