【数组】剑指 Offer 53 - I. 在排序数组中查找数字 I(简单)
程序员文章站
2023-12-28 13:32:58
...
【题目】
统计一个数字在排序数组中出现的次数。
【示例 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) {
map<int,int> m;
for(auto x:nums)
m[x]++;
return m[target];
}
};
【一行代码】
class Solution {
public:
int search(vector<int>& nums, int target) {
return upper_bound(nums.begin(),nums.end(),target)-lower_bound(nums.begin(),nums.end(),target);
}
};
【二分法】
class Solution {
public:
int search(vector<int>& nums, int target) {
int l=0,r=nums.size()-1,m=(l+r)/2,cnt=0;
if(r==-1)
return 0;
while(l<=r){
m=(l+r)/2;
if(nums[m]==target)
break;
else if(nums[m]>target){
r=m-1;
}else
l=m+1;
}
if(nums[m]==target){
int i=m;
while(i<nums.size()&&nums[i]==target){
cnt++;
i++;
}
i=m-1;
while(i>=0&&nums[i]==target){
cnt++;
i--;
}
}
return cnt;
}
};
【二分法简化版】
class Solution {
public:
int binarySearch(vector<int> nums,int target){
int i = 0, j = nums.size() - 1;
while(i <= j) {
int m = (i + j) / 2;
if(nums[m] <= target) i = m + 1;
else j = m - 1;
}
return i;
}
int search(vector<int>& nums, int target) {
return binarySearch(nums, target) - binarySearch(nums, target - 1);
}
};
推荐阅读