统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
二分法 1
排序数组 nums中的所有数字 target 形成一个窗口,记窗口的 左 / 右边界 索引分别为 left和 right,分别对应窗口左边 / 右边的首个元素。
本题要求统计数字 target的出现次数,可转化为:使用二分法分别找到 左边界 left和 右边界 right,易得数字 target的数量为 right−left−1
代码
class Solution {
public int search(int[] nums, int target) {
// 搜索右边界 right
int i = 0, j = nums.length - 1;
while(i <= j) {
int m = (i + j) / 2;
if(nums[m] <= target) i = m + 1;
else j = m - 1;
}
int right = i;
// 若数组中无 target ,则提前返回
if(j >= 0 && nums[j] != target) return 0;
// 搜索左边界 right
i = 0; j = nums.length - 1;
while(i <= j) {
int m = (i + j) / 2;
if(nums[m] < target) i = m + 1;
else j = m - 1;
}
int left = j;
return right - left - 1;
}
}
二分法 2优化
二分查找右边界 right的代码封装至函数
函数旨在查找数字 tar 在数组 nums中的 插入点 ,且若数组中存在值相同的元素,则插入到这些元素的右边。
由于数组 nums中元素都为整数,因此可以分别二分查找 target和 target−1的右边界,将两结果相减并返回即可。
代码
class Solution {
public int search(int[] nums, int target) {
//findRightPos:寻找x的有边界
//查找数字x在nums中的插入点
//若数组中存在值相同的元素,则插入到这些元素的右边
return findRightPos(nums,target)-findRightPos(nums,target-1);
}
public int findRightPos(int[] nums,int target){
int left=0;
int right=nums.length-1;
int mid;
//=时仍要判断与x大小
while(left<=right){
mid=left+(right-left)/2;
if(nums[mid]<=target){
left=mid+1;
}else{
right=mid-1;
}
}
return left;
}
}