题目:
统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
限制:
0 <= 数组长度 <= 50000
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/zai-pai-xu-shu-zu-zhong-cha-zhao-shu-zi-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
二分查找:
首先找到右边界,再找到左边界,最后取他们的差为结果。利用二分查找来达到这个目的。
在查找左边界时,设当前左下标为l,右下标为r,mid = (l+r)/2. 当mid的值小于目标值的时候,那么挪动l到mid+1,否则挪动r = mid。因为这时mid的值是大于等于目标的,也就是说mid右边是不会存在左边界的。跳出循环后检查nums[l]是否等于目标值,如果不符合,直接返回0,如果符合,存储l为左边界。
在查找右边界时,设mid为(l+r+1)/2, 这里的+1是为了在取半时向上取整。和查找左边界类似,当mid值大于目标值时,挪动r=mid-1。否则nums【mid】<= target, 也就是说mid左边是不存在右边界的,挪动l=mid。跳出循环后检查nums[r]是否等于目标值。如果不等于,返回0,否则存储r为右边界。最后取r-l+1为结果.
代码:
class Solution {
public:
int search(vector<int>& nums, int target) {
if (nums.empty())
{
return 0;
}
int l = 0;
int r = nums.size()-1;
int first = 0;
int last = 0;
// find left boundary
while (l < r)
{
int mid = (l+r)/2;
if (nums[mid] < target)
{
l = mid+1;
}
else
{
r = mid;
}
}
if (nums[l] != target)
{
return 0;
}
first = l;
//find right
l = 0;
r = nums.size()-1;
while (l < r)
{
// take upper value
int mid = (l+r+1)/2;
if (nums[mid] > target)
{
r = mid-1;
}
else
{
l = mid;
}
}
if (nums[r] != target)
{
return 0;
}
last = r;
return last-first+1;
}
};
时间复杂度: O(logn*2) = O(logn)
参考: