二分搜索

1)在有序数组中确定num存在还是不存在

public class FindNumber {

    // 在一个有序数组中找是否存在num
    public static boolean exist(int[] arr,int num) {
        int l = 0, r = arr.length-1, m = 0;
        while (l <= r) {
            m = (l + r) / 2;
            if (arr[m] == num) {
                return true;
            } else if (num < arr[m]) {
                r = m - 1;
            } else {
                l = m + 1;
            }
        }
        return false;
    }

    // --------- 用对数器检查是否正确 ---------
    public static boolean right(int[] arr, int num) {
        for (int i = 0;i < arr.length;i++) {
            if (arr[i] == num) {
                return true;
            }
        }
        return false;
    }

    // 生成随机数组
    public static int[] randomArray(int n, int v) {
        int[] arr = new int[n];
        for (int i = 0;i < n;i++) {
            // (int)Math.random() ——》 [0,1)
            // (int)(Math.random() * v) ——》 0 1 2 ... v-1
            // (int)(Math.random() * v) + 1 ——》 1 2 3 ... v
            arr[i] = (int)(Math.random() * v) + 1;
        }
        return arr;
    }

    public static void main(String[] args) {
        int N = 100;
        int V = 1000;
        int testTime = 50000;
        System.out.println("测试开始");
        for (int i = 0;i < testTime;i++) {

            int n = (int)(Math.random() * N); // 生成 0 ~ N-1 之间的随机数
            int[] arr = randomArray(n,V); // 生成 数组长度: 0 ~ N-1 ,数据大小 1 ~ V
            Arrays.sort(arr);
            int num = (int) (Math.random() * V) + 1;

            if (right(arr, num) != exist(arr, num)) {
                System.out.println("出错了!");
            }
        }
        System.out.println("测试结束");
    }
}

2)在有序数组中找>=num的最左位置

m >= num 记答案 往左二分, m < num 不记答案 往右二分

public class FindLeft {

    // 有序数组中找>=num的最左位置
    public static int findLeft(int[] arr, int num) {
        int l = 0,r = arr.length-1,m = 0;
        int ans = -1;
        while (l <= r) {
            m = (l + r) / 2;
            if (arr[m] >= num) {
                   ans = m;
                   r = m - 1;
            }else {
                l = m + 1;
            }
        }
        return ans;
    }

    public static int right(int[] arr, int num) {
        for (int i = 0;i < arr.length;i++) {
            if (arr[i] >= num) {
                return i;
            }
        }
        return -1;
    }

    // 生成随机数组
    // 为了验证
    public static int[] randomArray(int n, int v) {
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = (int) (Math.random() * v) + 1; // 1 ~ v
        }
        return arr;
    }

    public static void main(String[] args) {
        int N = 100;
        int V = 1000;
        int testTime = 500000;
        System.out.println("测试开始");
        for (int i = 0;i < testTime;i++) {
            int n = (int)(Math.random() * N); // 0 ~ N-1
            int[] arr = randomArray(n,V);
            Arrays.sort(arr);
            int num = (int) (Math.random() * V) + 1; // 1 ~ V
            if (right(arr, num) != findLeft(arr, num)) {
                System.out.println("出错了!");
            }
        }
        System.out.println("测试结束");
    }
}

3)在有序数组中找<=num的最右位置

public class FindRight {

    // 有序数组中找<=num的最右位置
    public static int findRight(int[] arr, int num) {
        int l = 0,r = arr.length-1,m = 0;
        int ans = -1;
        while (l <= r) {
            m = (l + r) / 2;
            if (arr[m] <= num) {
                ans = m;
                l = m + 1;
            }else {
                r = m - 1;
            }
        }
        return ans;
    }

    public static int right(int[] arr, int num) {
        for (int i = arr.length - 1; i >= 0; i--) {
            if (arr[i] <= num) {
                return i;
            }
        }
        return -1;
    }

    public static int[] randomArray(int n, int v) {
        int[] arr = new int[n];
        for (int i = 0; i < n; i++) {
            arr[i] = (int) (Math.random() * v) + 1;
        }
        return arr;
    }


    public static void main(String[] args) {
        int N = 100;
        int V = 1000;
        int testTime = 500000;
        System.out.println("测试开始");
        for (int i = 0; i < testTime; i++) {
            int n = (int) (Math.random() * N);
            int[] arr = randomArray(n, V);
            Arrays.sort(arr);
            int num = (int) (Math.random() * V) + 1;
            if (right(arr, num) != findRight(arr, num)) {
                System.out.println("出错了!");
            }
        }
        System.out.println("测试结束");
    }
}

4)二分搜索不一定发生在有序数组上(比如寻找峰值问题) 

什么是峰值 ?

arr[i-1] < arr[i] > arr[i+1]

如何找到峰值 ?

1. 一个数默认就是峰值,因为假设 nums[-1] = nums[n] = -∞

2. 看数组第一个数和最后一个数是不是峰值

3. 如果都不是 就可以确定峰值在下标[1~n-2]之间,开始二分查找

为什么可以使用二分 ?

因为确定峰值就在l~r之间,一定可以找到

class Solution {
    public int findPeakElement(int[] nums) {
        // 如果就一个数 默认就是峰值 因为左右都是无穷小
        if (nums.length == 1) {
            return 0;
        }
        // >= 2
        if (nums[0] > nums[1]) {
            return 0;
        }
        int n = nums.length;
        if (nums[n-1] > nums[n-2]) {
            return n-1;
        }

        // l - r之间一定有峰值 
        int l = 1,r = n-2,m = 0,ans = -1;
        while (l <= r) {
            m = (l + r) / 2;
            if (nums[m-1] > nums[m]) {
                r = m - 1;
            }else if (nums[m+1] > nums[m]) {
                l = m + 1;
            }else { 
                ans = m;
                break;
            }
        }
        return ans;
    }
}

 

posted @ 2024-02-05 10:53  qyx1  阅读(25)  评论(0)    收藏  举报