统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
方法:二分法
首先由经典的二分法搜索得到nums中任意一个target的位置,之后分别向两边计数。
特殊情况:
- nums=[]nums = []nums=[],此时r=−1,l>rr = -1,l >rr=−1,l>r,直接跳到最下return0return 0return0。
- 只有一项nums[1],target=1nums[1],target = 1nums[1],target=1,此时第一次进入whilewhilewhile循环,nums[m]nums[m]nums[m]就是 =target= target=target的,直接开始计数,在计数中,ml=−1,mr=1m_l = -1,m_r = 1ml=−1,mr=1均超出索引范围,都不满足计数whilewhilewhile的条件,因此直接输出此时的n=1n = 1n=1。
C++
class Solution {
public:
int search(vector<int>& nums, int target) {
int r = nums.size()-1, l = 0, n = 1;
int m = 0;
while(l<=r){
m = (l+r)/2;
if(nums[m]<target) l = m+1;
else if(nums[m]>target) r = m-1;
else{
int m_l = m-1, m_r = m+1;
while(m_l >= 0 && nums[m_l] == target) {
n++;
m_l--;
}
while(m_r<=nums.size()-1 && nums[m_r] == target) {
n++;
m_r++;
}
return n;
}
}
return 0;
}
};