- 二分查找
简单
1.3K
相关企业
给定一个 n 个元素有序的(升序)整型数组 nums 和一个目标值 target ,写一个函数搜索 nums 中的 target,如果目标值存在返回下标,否则返回 -1。
示例 1:
输入: nums = [-1,0,3,5,9,12], target = 9
输出: 4
解释: 9 出现在 nums 中并且下标为 4
示例 2:
输入: nums = [-1,0,3,5,9,12], target = 2
输出: -1
解释: 2 不存在 nums 中因此返回 -1
二分查找模板题,不需具体考虑总数是奇数还是的具体情况,其思想就是根据其数据的有序,找到其中间数据来简化运算,因此我们在写此类题时要注意的是确保中间值取值正确,和什么时候跳出循环。
class Solution {
public:
int search(vector<int>& nums, int target) {
int right=nums.size()-1,left=0;
while(left<=right)
{
int mid=(left+right)/2;
if(nums[mid]>target)
{
right=mid-1;
}
else if(nums[mid]<target)
{
left=mid+1;
}
else
{
nums[mid]==target;
return mid;
}
}
return -1;
}
};
剑指 Offer 53 - I. 在排序数组中查找数字 I
简单
418
相关企业
统计一个数字在排序数组中出现的次数。
示例 1:
输入: nums = [5,7,7,8,8,10], target = 8
输出: 2
示例 2:
输入: nums = [5,7,7,8,8,10], target = 6
输出: 0
这个题也用二分查找,只不过查找的的目标值有所不同,并对结果稍加处理。
我们统计一个数字在排序数组中出现的的次数,只需将大于此数的最小位置减去该数的出现的最初的位置即可。因此我们可以定义一个函数来返回大于等于某一数据的最小位置。在这个函数中使用二分查找是要注意:之前nums[mid]>target时,我们right=mid-1,而现在应该是if(nums[mid]>=target)
{
r=mid-1;
ans=mid;
}
class Solution {
public:
int bSearch(vector<int> & nums,int n,int target)
{
int l=0,r=n-1;
int ans=n;
while(l<=r)
{
int mid=(l+r)>>1;
if(nums[mid]>=target)
{
r=mid-1;
ans=mid;
}
else if(nums[mid]<target)
{
l=mid+1;
}
}
return ans;
}
int search(vector<int>& nums, int target) {
int n=nums.size();
int index=bSearch(nums,n,target);
if(index==n||nums[index]!=target)
{
return 0;
}
return bSearch(nums,n,target+1)-index;
}
};