java实现 lowerBound 和 upperBound

305 阅读1分钟

c++中 lower_bound 和 upper_bound非常经典,我们用java实现一下。

 /**
     * @param arr
     * @param value
     * @return 第一个大于等于value的数的坐标
     */
    int lowerBound(int[] arr, int value) {
        int l = 0, r = arr.length - 1;
        while (l <= r) {
            int m = ( l+r) / 2;
            if (arr[m] < value) {  
                l = m + 1;     // 如果m位置的元素太小,直接把左边界跳到m+1
            } else { // 相当于 arr[m] >= value
                r = m - 1;  // 虽然m有可能是目标解,直接m-1会错过,但是最后如果在 l 和 m -1 里面找不到, l会取“m+1”,跳出循环,这里的“m+1“其实就是错过的目标解、
            }
        }
        return l;
    }

    /**
     * @param arr
     * @param value
     * @return 第一个大于value的数的坐标
     */
    int upperBound(int[] arr, int value) {
        int l = 0, r = arr.length - 1;
        while (l <= r) {
            int m = ( l+r) / 2;
            if (arr[m] <= value) {
                l = m + 1;
            } else {  // arr[m] > value
                r = m - 1;
            }
        }
        return l;
    }

本文已参与「新人创作礼」活动,一起开启掘金创作之路