[剑指offer]在排序数组中查找数字I
剑指offer-在排序数组中查找数字I
统计一个数字在排序数组中出现的次数。
题目描述
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
限制:
0 <= 数组长度 <= 50000
解题思路
- 面试官出这个题大概率是想看你写二分
- 用两次二分,分别找这个数字出现的右边界和左边界。
实现代码
class Solution {
public:
int search(vector<int>& nums, int target) {
int len=nums.size();
if(len==0)
return 0;
int count=0;
int i=0,j=len-1;
while(i<=j){//第一次二分查找,找右边界
int mid=(i+j)/2;
if(nums[mid]<=target)
i=mid+1;
else
j=mid-1;
}
if(j>0&&nums[j]<target)//数组中没有target,可以提前出循环
return 0;
int right=j;
i=0;
while(i<=j){//第二次二分查找,找左边界
int mid=(i+j)/2;
if(nums[mid]<target)
i=mid+1;
else
j=mid-1;
}
int left=i;
return right-left+1;
}
};