题目
统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
限制:
- 0 <= 数组长度 <= 50000
解题
主要思路:有序的数组查找,为了减少时间复杂度肯定需要使用二分查找;但是不是找元素的位置,而是查找元素出现的次数;可以采用常规的二分查找:
- 当中间元素大于target,那么右边界为middle-1;
- 当中间元素小于target,那么左边界为middle+1;
- 当中间元素等于target,那么左边界是一个区间,右边界也是一个区间;
正常考虑会复杂许多,此时可以使用二分查找的递归方式,当中间元素等于target,再对左右区间查找target。
时间复杂度:O(LogN) N为数组的长度
空间复杂度:O(1)
int count = 0;
public int search(int[] nums, int target) {
helper(nums, 0, nums.length-1, target);
return count;
}
void helper(int[] nums, int low, int high, int target) {
if (low <= high) {
int middle = low + (high - low)/2;
if (nums[middle] == target) {
count++;
helper(nums, low, middle-1, target);
helper(nums, middle+1, high, target);
} else if (nums[middle] > target) {
helper(nums, low, middle-1, target);
} else {
helper(nums, middle+1, high, target);
}
}
}