统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
【解题思路】
由于是有序数组,考虑采用二分法,分别查找数组的左边界和右边界,在使用二分法的过程中,注意是否有+1的情况,具体可以参考如下的二分模板:
代码如下:
class Solution {
public int search(int[] nums, int target) {
if(nums == null||nums.length == 0){
return 0;
}
int l = 0, r = nums.length - 1;
while(l < r){
int mid = l + r >> 1;
if(nums[mid] >= target){
r = mid;
}else{
l = mid + 1;
}
}
int left = l;
l = 0;
r = nums.length - 1;
while(l < r){
int mid = l + r + 1 >> 1;
if(nums[mid] <= target){
l = mid;
}else{
r = mid - 1;
}
}
int right = r;
if(nums[left] != target){
return 0;
}
return right - left + 1;
}
}