二分查找之寻找左右边界

715 阅读1分钟

class Main {

int left_bound(int[] nums, int target) {
    if (nums.length == 0) {
        return -1;
    }
    int left = 0;
    int right = nums.length;

    while(left < right) { // 搜索区间[left, right)左闭右开, 结束条件:left==right
        int mid = left + (right - left) / 2;
        if (nums[mid] == target) {
            right = mid;
        } else if (nums[mid] > target) {
            right = mid;
        } else if (nums[mid] < target) {
            left = mid + 1;
        }
    }

    // left 取值区间[0, right]
    // 如果target比nums中任何数都要大,那么left会一直右移来缩小区间,直到left==right==nums.length
    // 如果target比nums中任何数都要小,那么right会一直左移来缩小区间,直到right==left==0
    if(left == nums.length) {
        return -1;
    }
    // 结束条件:left==right,所以返回时用left或者right都可以
    return nums[left] == target ? left : -1;
}


int right_bound(int[] nums, int target) {
    if (nums.length == 0) {
        return -1;
    }
    int left = 0;
    int right = nums.length;

    while (left < right) {
        int mid = left + (right - left) / 2;
        if (nums[mid] == target) { // 只要nums中存在任意target,那么搜索结束时left指针就不可能为0
            left = mid + 1;
        } else if (nums[mid] > target) {
            right = mid;
        } else if (nums[mid] < target) {
            left = mid + 1;
        }
    }

    // 只要nums中存在任意target,那么搜索结束时left指针就不可能为0
    if (left == 0) {
        return -1;
    }

    // left-1是因为每次遇到nums[mid] == target时,都赋值left=mid+1;
    // 所以判断nums[left-1]来确定右边界
    return nums[left - 1] == target ? left - 1 : -1;
}

}