NC105_二分查找

知识点：二分查找
题目链接

题目描述
请实现有重复数字的升序数组的二分查找。
输出在数组中第一个大于等于查找值的位置，如果数组中不存在这样的数(指不存在大于等于查找值的数)，则输出数组长度加一。

输入：
5,4,[1,2,4,4,5]
输出：
3

备注：
输出位置从1开始计算

解题思路

1. 找到什么样的一定不是结果 如果比v小 一定不是结果 left = mid+1
2. 如果大于等于v 因为存在重复数字 所以还是要往左边找

代码

#include "cheader.h"
class Solution {
public:
    int upper_bound_(int n, int v, vector<int>& a) {
        int left = 0,right = n-1;
        int flag = 0;
        while(left < right){
            int mid = left + (right-left)/2;
            if(a[mid] >= v){
                flag = 1;
                right = mid;
            }
            else if(a[mid] < v)
                left = mid+1;
        }
        return flag==1?left+1:n+1;
    }
};

int main()
{
    vector<int> a{1,2,4,4,5};
    Solution s;
    cout<<s.upper_bound_(5, 4, a)<<endl;
    return 0;
}

今天也是爱zz的一天