二分查找——如何写题更快

对于二分查找我们可以有很多变形的题来做，那万变不离其宗的还是最原始的二分查找。在此基础上我们进行一些操作去完成更加复杂的题目。接下来给大家分享分享。

模板

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

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

例题1

你是产品经理，目前正在带领一个团队开发新的产品。不幸的是，你的产品的最新版本没有通过质量检测。由于每个版本都是基于之前的版本开发的，所以错误的版本之后的所有版本都是错的。

假设你有 n 个版本 [1, 2, …, n]，你想找出导致之后所有版本出错的第一个错误的版本。

你可以通过调用 bool isBadVersion(version) 接口来判断版本号 version 是否在单元测试中出错。实现一个函数来查找第一个错误的版本。你应该尽量减少对调用 API 的次数。

力扣（LeetCode）
链接：https://leetcode-cn.com/problems/first-bad-version

我们审题便不难发现它是一个二分查找的题，由此我们可以加上一些判断条件即可
代码如下

class Solution {
public:
    int firstBadVersion(int n) {
        int  start = 1;
        int end = n;
        int middle;
        while(start<=end){
            middle =start+(end-start)/2; 
            if(isBadVersion(middle) == true && isBadVersion(middle-1) == false)
                break;
            else if( isBadVersion(middle) == true){
                 end = middle-1;
            }else{
                start = middle+1;
            }
        }
        return middle;
    }
};

例题2

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-insert-position

我们要判断插入的位置，这块要注意，假设最终我们没有找到值，那如何插入进去是我们该思考的。
我们要注意此处的middle与target的关系，如果middle小于target，那应该插入后一位，middle>target，则应该插入middle的位置。当然如果等于则在前面已经找到了。
代码

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

总结

你会发现这两道变形题最基本的还是原始算法，所以掌握好最基础的模板，是比较重要的。这样我们在写类似的题目时，可以更快先人一步。