【剑指Offer系列53-1】 在排序数组中查找数字 I

题目

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

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

示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0

限制：
0 <= 数组长度 <= 50000

代码

Python

# 思路：
# 采用二分查找法
# 复杂度：
# O(logN)
class Solution:
    def search(self, nums: List[int], target: int) -> int:
        def helper(tar):
            i,j=0,len(nums)-1
            while i<=j:
                m=(i+j)//2
                if nums[m]<=tar: i=m+1
                else: j=m-1
            return i
        return helper(target)-helper(target-1)

C++

class Solution {
public:
    int search(vector<int>& nums, int target) {
        return helper(nums,target)-helper(nums,target-1);
    }

    int helper(vector<int>& nums,int target) {
        int i=0,j=nums.size()-1;
        while (i<=j) {
            int m=(i+j)/2;
            if (nums[m]<=target) i=m+1;
            else j=m-1;
        }
        return i;
    }
};