704. 二分查找(C++)

题目

给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。

示例 1:

输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4

示例 2:

输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1

提示:

  • 你可以假设 nums 中的所有元素是不重复的。
  • n 将在 [1, 10000]之间。
  • nums 的每个元素都将在 [-9999, 9999]之间。

分析与题解

在循环体内部查找元素

  • while(left <= right) 这种写法表示在循环体内部直接查找元素
  • 退出循环的时候 left 和 right 不重合,区间[left, right]是空区间。

此时相当于判断区间被分为三节,分别为[left , mid-1],[mid],[mid+1 , right](从mid部分被左右进行切分),代码如下:

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int len = nums.size();
        if(len==0)
            return -1;
        int left=0,right=len-1;
        while(left<=right){
            //防止left和right过大造成整型溢出
            //在面对left=mid,right=mid-1永远取不到右边界
            //边界收缩行为有可能产生死循环
            int mid = left + (right-left)/2;
            if(nums[mid]==target)
                return mid;
            else if(nums[mid]<target)
                //搜索区间为 [mid+1 , right]
                left = mid + 1;
            else
                //搜索区间为 [left , mid-1]
                right = mid - 1;
            }
        return -1;
    }
};

在循环体内部排除元素

  • while(left < right) 这种写法表示在循环体内部排除元素;

  • 退出循环的时候 left 和 right 重合,区间[left, right]只剩下成 11 个元素,这个元素有可能就是我们要找的元素。

    left、right重合时的情况再循环内没有判断,所以跳出循环后仍需要进行一次判断

首先,先判断答案可以的位置:

  • 插入位置有可能在数组的末尾,需要单独判断。如果在数组的末尾,插入位置的下标就是数组的长度;
  • 否则,根据示例的分析,插入位置的下标是 大于等于 target 的第 11 个元素的位置。

因此,严格小于 target 的元素一定不是解,在循环体中将左右边界 left 和 right 逐渐向中间靠拢,最后 left 和 right 相遇,则找到了插入元素的位置。根据这个思路,可以写出如下代码。

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int len = nums.size();
        if(len==0)
            return -1;
        int left=0,right=len-1;
        while(left<right){
            int mid = left + (right-left)/2;
            //直接回避上取整的问题
            if(nums[mid]<target)
                left = mid+1;
            else
                right = mid;
        }
        if(nums[left]==target)
            return left;
        else
            return -1;
    }
};

因为题解中对边界的处理是左动,因此遵循下取整,不需要对中值求值进行修改。

posted @ 2020-08-25 17:38  脱线森林`  阅读(127)  评论(0编辑  收藏  举报