给定一个 n
个元素有序的(升序)整型数组 nums
和一个目标值 target
,写一个函数搜索 nums
中的 target
,如果目标值存在返回下标,否则返回 -1
。
1.从头到尾采用for循环遍历,比较每个元素与target
class Solution {
public:
int search(vector<int>& nums, int target) {
int len = nums.size();
int index = -1;
for(int i = 0;i < len; i++){
if(nums[i]==target){
index = i;
}
}
return index;
}
};
2.二分查找
这道题目的前提是数组为有序数组,同时题目还强调数组中无重复元素,因为一旦有重复元素,使用二分查找法返回的元素下标可能不是唯一的,这些都是使用二分法的前提条件
注意区间,代码采用左闭右闭区间,这样循环条件为:left<=right
原理大意为:
由于是升序数组,且无重复,那么可以考虑比较区间中间的元素值与target的大小,
如果target小,那么要找的位置一定在中间的左侧,那么新的搜索区间为[left,middle]
如果target大,那么要找的位置一定在中间的右侧,那么新的搜索区间为[middle,right],
跟据这两点更新区间,这样减小搜索范围,继续比较新区间中间元素与target的大小,如此循环,直到找到位置,如果不存在,return -1
(具体可以看一下b站这个视频,视频更清晰一些,5分钟学会:二分查找_哔哩哔哩_bilibili)
代码:
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0, right = nums.size() - 1;
while(left <= right){
int mid = (right - left) / 2 + left;
int num = nums[mid];
if (num == target) {
return mid;
} else if (num > target) {
right = mid - 1;
} else {
left = mid + 1;
}
}
return -1;
}
};
作者:力扣官方题解
链接:https://leetcode.cn/problems/binary-search/solutions/980494/er-fen-cha-zhao-by-leetcode-solution-f0xw/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
类似题型:
35. 搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n)
的算法
将上面的代码稍作改进即可
最后只要返回left就是题目要求的,可以自己画图体会一下
根据if的判断条件,left左边的值一直保持小于target,right右边的值一直保持大于等于target,而且left最终一定等于right+1,这么一来,循环结束后,在left和right之间画一条竖线,恰好可以把数组分为两部分:left左边的部分和right右边的部分,而且left左边的部分全部小于target,并以right结尾;right右边的部分全部大于等于target,并以left为首。所以最终答案一定在left的位置。
class Solution {
public:
int searchInsert(vector<int>& nums, int target) {
int n = nums.size();
int left = 0,right = n-1;
while(left <= right){
int mid= left + (right - left)/2;
if(nums[mid]<target){
left = mid + 1;
}else{
right = mid - 1;
}
}
return left;
}
};
34. 在排序数组中查找元素的第一个和最后一个位置(中等题)
给你一个按照非递减顺序排列的整数数组 nums
,和一个目标值 target
。请你找出给定目标值在数组中的开始位置和结束位置。
如果数组中不存在目标值 target
,返回 [-1, -1]
。
你必须设计并实现时间复杂度为 O(log n)
的算法解决此问题。
官方的答案很复杂,其实使用两次二分查找好了,虽然代码长一些,但是很好理解
class Solution {
public:
vector<int> searchRange(vector<int>& nums, int target) {
int n = nums.size();
if (n == 0) return vector<int>{-1, -1};
int l = 0, r = n - 1;
while (l < r) { // 寻找左边界
int mid = (l + r) >> 1;//>>比特操作,相当于除2
if (nums[mid] < target) l = mid + 1;
else r = mid;
}
if (nums[l] != target) return vector<int>{-1, -1};
int x = l;
l = 0, r = n - 1;
while (l < r) { // 寻找右边界
int mid = (l + r + 1) >> 1;
if (nums[mid] > target) r = mid - 1;
else l = mid;
}
return vector<int>{x, l};
}
};
69 x 的平方根
给你一个非负整数 x
,计算并返回 x
的 算术平方根 。
由于返回类型是整数,结果只保留 整数部分 ,小数部分将被 舍去 。
注意:不允许使用任何内置指数函数和算符,例如 pow(x, 0.5)
或者 x ** 0.5
。
class Solution {
public:
int mySqrt(int x) {
int left = 0;
int right = x;
while(left <= right){
int mid = (left + right)/2;
if((long long)mid*mid < x){
left = mid + 1;
}else if((long long)mid*mid > x){
right = mid - 1;
}else{
return mid;
}
}
return right;
}
};
367. 有效的完全平方数
给你一个正整数 num
。如果 num
是一个完全平方数,则返回 true
,否则返回 false
。
完全平方数 是一个可以写成某个整数的平方的整数。换句话说,它可以写成某个整数和自身的乘积。
不能使用任何内置的库函数,如 sqrt
。
和上题是一样的思路
class Solution {
public:
bool isPerfectSquare(int num) {
int left = 0, right = num;
while (left <= right) {
int mid = (right - left) / 2 + left;
long square = (long) mid * mid;
if (square < num) {
left = mid + 1;
} else if (square > num) {
right = mid - 1;
} else {
return true;
}
}
return false;
}
};