剑指 Offer 53 - I. 在排序数组中查找数字 I

题目

统计一个数字在排序数组中出现的次数。

示例

输入: nums = [5,7,7,8,8,10], target = 8
输出: 2

代码一

遍历数组，直接进行统计就结束了。O(n)

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int tot = 0;
        for(int i = 0;i < nums.size();i++){
            if(nums[i] == target) tot++;
        }
        return tot;
    }
};

代码二

上面那种也太简单了。所以本质上这道题是用二分找第一个大于target数组的位置和找第一个大于target-1的位置。有序数组肯定用的是o(logn)的二分。

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