leetcode-剑指 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 <= 数组长度 <= 50000
解题思路:
方法一: 常规思考方式是遍历nums数组,遇到target就将计数cnt++,最后返回cnt.
时间复杂度: O(n)
空间复杂度:O(1)
class Solution {
public:
int search(vector<int>& nums, int target) {
int cnt=0;
for(int i=0;i<nums.size();i++)
{
if(nums[i]==target)
cnt++;
}
return cnt;
}
};
方法二: 可以使用二分方法降低时间复杂度。
分别求出target在nums数组中第一次出现、最后一次出现的下标,也就是求距离。
在两次求端点的过程中:
共同点
a、当nums[mid] > target 时 ,right=mid-1;
b、当nums[mid] < target 时 ,left=mid+1;
如果if (nums[mid] == target )时左右端点的计算有 区别
要确定左端点的值,所以缩小范围,不断左移右端点的值: right–;
要确定右端点的值,所以缩小范围,不断右移动左端点的: left++;
然后每次标记位置
class Solution {
public:
int search(vector<int>& nums, int target) {
int left=0,right=nums.size()-1;
int l=0,r=0;
while(left<=right) //找右端点
{
int mid=left+(right-left)/2;
if(nums[mid]>target)
right=mid-1;
else if(nums[mid]<target)
left=mid+1;
else
{
left++;
r=left; //返回的是target最后一次出现的下一个位置
}
}
left=0,right=nums.size()-1;
while(left<=right) //找左端点
{
int mid=left+(right-left)/2;
if(nums[mid]>target)
right=mid-1;
else if(nums[mid]<target)
left=mid+1;
else
{
right--;
l=left; //返回的是target第一次出现的位置
}
}
return r-l;
}
};