给定一个 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提示:你可以假设 nums 中的所有元素是不重复的。n 将在 [1, 10000]之间。nums 的每个元素都将在 [-9999, 9999]之间。
思路
二分查找,简单来说,就是先用数组中间位置的值与目标值比较,如果相等,则查找成功,如不相等,则利用中间位置将数组分成前后两部分,对于升序的数组来说,目标值大于中间位置的值,则进一步查找后半部分的中间位置,否则查找前半部分的中间位置,重复以上过程直到查找成功或数组不可再分为止,此时查找失败。每次都能排除一半的元素,时间复杂度只有 O ( log n ) O(\log n) O(logn),是一种效率很高的算法。
逻辑虽然简单,但是要写起来却很容易出错,因为涉及很多边界条件。
首先要注意使用二分查找的条件:数组中元素是有序的,数组中元素是不重复的。
二分法主要有两种写法,最主要是确定区间范围,一种是左闭右闭[left, right]
初始化right = nums.size() - 1
,一种是左闭右开[left, right)
初始化right = nums.size()
下面就这两种写法进行讲解
写法一:左闭右闭
C++代码如下:
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;
if (nums[mid] == target) {
return mid;
}
else if (nums[mid] > target) {
right = mid - 1;
}
else {
left = mid + 1;
}
}
return -1;
}
};
这种写法主要记住两点:
while (left <= right)
要用<=
,因为左闭右闭对于区间内只有一个值也是有意义的。if (nums[mid] > target) right = mid - 1
,因为当前的nums[mid]
一定不是target
,所以right
要赋值mid - 1
如要查找的target
为1
这种情况下,如果将<=
写成了<
那么在left == right
的时候就退出了循环,mid
也就不会指向1
,最终导致未找到的错误结果。(上图是正确的情况)
写法二:左闭右开
这种写法也是记住两点:
while (left < right)
要用<
,因为左闭右开对于区间内只有一个值是没有意义的。if (nums[mid] > target) right = mid
因为当前的nums[mid]
一定不是target
,而因为是右开,从right - 1
开始向前才是接下来要查找的元素,right
要赋值为mid
。
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0, right = nums.size();
while (left < right) {
int mid = (right - left) / 2 + left;
if (nums[mid] == target) {
return mid;
}
else if (nums[mid] > target) {
right = mid;
}
else {
left = mid + 1;
}
}
return -1;
}
};
如要查找的元素为3
将right = mid写成了right = mid - 1那么就会出现如下错误。3直接错过了。
总结
二分法是非常重要的基础算法,它很简单,但是总容易一学就会,一写就废,为什么呢?关键就在于把握好区间的控制。
理解好这篇文章所说的几点,相信以后遇到二分法就能轻松熟练有把握地写出来了吧。😊