剑指 Offer 53 - I. 在排序数组中查找数字 I
统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
思路一:可以直接遍历数组,记录目标值的个数即可。
思路二:我们观察到题目给到的数组是一个已经有序的数组,故我们可以考虑采用二分法分别来查找最左以及最右的目标值的索引,最后进行相减即可,但是我们知道二分法只是负责找到那个目标值而已,并没有办法找到最左或者最右的目标值。所以我们需要对二分法进行一点小优化,即找到目标值时,不着急返回,而是继续往左或者往右缩小区间。
class Solution {
public int search(int[] nums, int target) {
int left = 0;
int right = nums.length;
// 先找左边,终止条件为left = right时
while(left < right){
int mid = left + (right - left) / 2;
if(nums[mid] < target) left = mid+1;
// 这里其实可以将两个else合并成一个,但是为了逻辑看起来更清晰,多写了一个else,查找右边的时候同
else if(nums[mid] > target) right = mid;
else right = mid;
}
int temp1 = left;
if(left >= nums.length || nums[left] != target) return 0;
// 找右边
left = 0;
right = nums.length;
while(left < right){
int mid = left + (right - left) / 2;
if(nums[mid] < target) left = mid+1;
else if(nums[mid] > target) right = mid;
else left = mid+1;
}
// 注意,在查找右边的时候,实际返回的应该时left-1,
//因为我们找到目标的时候left = mid+1,循环结束的时候left = right
//此时nums[left-1]才等于target,这也是为什么在return的时候不需要+1
//通常索引返回个数需要+1,而left = mid+1已经加过了 故省略了,
//真正的返回值应该是如下表示更清晰 right - 1 - temp1 + 1;
return right - temp1;
}
}