【二分查找+搜索插入位置】二分查找法初探

一、题目

二、解法

上述问题的关键在于升序排序的数组，因此可以不去使用暴力解法进行遍历，而是采用二分查找。具体来说，

• 首先确定初始查找范围，即整个数组（查找索引left ~ right为0 ~ 数组长度-1）；
• 在循环中，每次选择查找范围的中间元素（索引为mid = left + (right - left) / 2，注意取整即可）和给定元素target进行比较；
• 由于是升序数组，因此如果中间元素>target，则target一定在left ~ mid-1之间，即更新right为mid-1，若中间元素<target，则target一定在mid+1~ right之间，即更新left为mid+1，若中间元素=target，即返回mid索引即为结果；

• 对于第一个问题，到此即可解答完成，也就是说循环过程中能找到中间元素=target，那么直接返回mid，否则就结束循环返回-1;
• 需要特别注意的是，结束循环的条件为left <= right，即left==right时也需要进入循环，mid=left=right，也需要判断该位置的元素是否等于target；
• 例如，left和right二者相差1时，进入循环后，mid=left，此时可能有三种情况，left的元素=target，直接返回mid；left元素<target，更新left为mid+1，此时left和right相等，而这时候还需要判断right元素和target元素是否相等，则还得进入一次循环，循环条件必须为left <= right；left的元素>target，则right元素肯定也大于target，target肯定不存在；

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int n = nums.size();
        int left = 0, right = n-1;
        while (left<=right){
            int mid = left + (right-left)/2;
            if (nums[mid]<target){
                left = mid+1;
            }
            else if (nums[mid]>target){
                right = mid-1;
            }
            else{
                return mid;
            }
        }
        return -1;
    }
};

• 对于第二个问题，只是多了要返回一个插入的位置，可以这样思考，如果数组中没有target，最后一次循环肯定是left=right=mid，因为在不断的对查找范围进行二分，此时有两种情况：
• 该位置的元素<target，left = mid+1，target应该插在该元素之后，即返回left即可；
• 该位置的元素>target，right = mid-1，target应该插在该元素的位置，也应该返回left。

class Solution {
public:
    int searchInsert(vector<int>& nums, int target) {
        int n = nums.size();
        int left = 0, right = n-1;
        while (left <= right){
            int mid = left + (right - left)/2;
            if (nums[mid] < target){
                left = mid+1;
            }
            else if (nums[mid] > target){
                right = mid-1;
            }
            else{
                return mid;
            }
        }
        return left;
    }
};

综上，二分查找尤其需要考虑的是循环结束条件以及返回值。