题目链接:剑指 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
提示:
- 0 <= nums.length <= 105
- -109 <= nums[i] <= 109
- nums 是一个非递减数组
- -109 <= target <= 109
注意:本题与主站 34. 在排序数组中查找元素的第一个和最后一个位置相同(仅返回值不同),34题题解见:力扣:34. 在排序数组中查找元素的第一个和最后一个位置。
思路和算法:
在这题中,主要关注的点是这个数组是排序数组。那么,虽然双指针能完成查找符合的区间,但是时间复杂度为O(n),并且也没有用上题中给的排序这个条件。显然这个时候用二分查找会比较快,并且时间复杂度也仅为O(nlog n)。
寻找区间左边界和右边界,用的二分查找有部分代码不一样,但是大部分都是可以复用的,这时就可以将代码拿出来单独写,复用代码条件需要加上一个标记(flag)来区分是寻找左边界还是右边界。
代码(c++):
//二分查找:左闭右开区间
class Solution {
public:
//抽出可复用的代码
//flag标记是寻找左边界还是右边界
int binarySearch(vector<int>& nums, int target, bool flag) {
int left = 0, right = nums.size() - 1, ans = nums.size();
while (left <= right) {
int mid = left + (right - left) / 2;
if (nums[mid] > target || (flag && nums[mid] >= target)) {
right = mid - 1;
ans = mid;
}
else {
left = mid + 1;
}
}
return ans;
}
int search(vector<int>& nums, int target) {
//寻找与目标值相等的左边界
int leftIndex = binarySearch(nums, target, true);
//寻找大于目标值的最小右边界
int rightIndex = binarySearch(nums, target, false);
if (leftIndex < rightIndex && rightIndex <= nums.size() && nums[leftIndex] == target && nums[rightIndex - 1] == target) {
return rightIndex - leftIndex; //返回等于目标值的区间
}
return 0; //不存在则返回 0
}
};