【五月集训】（第九天）—— 二分查找

---

解题报告

搜索插入位置

题目链接：35. 搜索插入位置

解题思路

二分查找，l 从下标0开始，r 从数组长度 + 1开始（因为这也可能是答案）。
保证 l < r，进行折半查找，如果nums[mid] >= target，区间缩小为[l, mid]，注意右边界是可能取到的；如果nums[mid] < target，区间缩小为[mid + 1, r]，跳过了 mid。

解题代码

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

二分查找

题目链接：704. 二分查找

解题思路

二分查找，类似上一题思路，但要注意是l <= r，因为最后两个指针指向同一个下标时要判断这个数是否为target。
折半查找，如果nums[mid] == target，说明找到了，返回mid，否则不断缩减查找区间，直到l > r，说明没找到，返回 -1。

解题代码

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

在排序数组中查找数字 I

题目链接：剑指 Offer 53 - I. 在排序数组中查找数字 I

解题思路

主要是找到两个下标，目标数字的起始下标和结束下标。可以使用二分查找到target的起始下标，和target + 1的下标，target + 1的下标即为大于target的最小下标，也就是最后一个target的下标 + 1的位置。
二分查找的方法与第一题一模一样，注意r = nums.size()，因为要找target + 1，例如[1,3,5]要找5，target + 1得到的下标应该为 3。

解题代码

class Solution {
public:
    int bsearch(vector<int>& nums, int target){
        int l = 0, r = nums.size();
        while(l < r){
            int mid = (l + r) >> 1;
            if(nums[mid] >= target){
                r = mid;
            }else{
                l = mid + 1;
            }
        }
        return l;
    }

    int search(vector<int>& nums, int target) {
        if(nums.size() == 0)
            return 0;
        return bsearch(nums, target + 1) - bsearch(nums, target);
    }
};

在线选举

题目链接：911. 在线选举

解题思路

预处理一个 tops 数组用来存放投完第 i 次票后，处于领先的候选人编号。q 函数用来查询第 i 个时刻处于领先的候选人编号。
（1）tops[i]存放投完第 i 次票后，处于领先的候选人编号；_times存放时刻表。
（2）sum用来记录每个候选人的得票。
（3）maxV记录当前最大得票数，maxP记录当前最大得票数的候选人编号。
（4）~（6）遍历person数组，累计每个人的得票数，并预处理每次投票后的领先人数组tops。
（5）注意处理最大得票数要用>=，因为题意中“在平局的情况下，最近获得投票的候选人将会获胜”。
（7）二分查找小于等于 t 时刻的下标，该下标对应是第几次投票，返回tops[l]，即该时刻投完票后的领先人。

解题代码

class TopVotedCandidate {
public:
    vector<int> tops;				//(1)
    vector<int> _times;

    TopVotedCandidate(vector<int>& persons, vector<int>& times) {
        int n = persons.size();
        tops.resize(n);
        _times = times;
        int sum[n];						//(2)
        memset(sum, 0, sizeof(int) * n);
        
        int maxV = 0;					//(3)
        int maxP;
        for(int i = 0; i < n; i ++ ){	//(4)
            int p = persons[i];
            sum[p] ++ ;
            if(sum[p] >= maxV){			//(5)
                maxV = sum[p];
                maxP = p;
            }
            tops[i] = maxP;				//(6)
        }
    }
    
    int q(int t) {
        int l = 0, r = _times.size() - 1;	//(7)
        while(l < r){				
            int mid = (l + r + 1) >> 1;
            if(_times[mid] <= t){
                l = mid;
            }else{
                r = mid - 1;
            }
        }
        return tops[l];
    }
};