leetcode704.二分查找
1 题目
给定一个 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]之间。
2 思路
注意对题目中有序数组,数组中无重复元素这两个keywords。
这些都是使用二分法的前提条件。因为若出现重复元素,返回的下标便不唯一了。
虽然很轻松就通过了但是还是有一些细节需要注意。例如我第一次提交的时候边界条件写成了while ( low != high ) 乍一看好像没问题,特殊情况就在于,如果给定数组nums中只有一个元素的时候,就进入不了循环最后返回-1。
对于二分查找的边界条件还是有一些细节要注意。这里学习到了Carl代码随想录中非常细节的地方。Carl代码随想录本题链接
也就是二分法区间的两种定义:
2.1 二分法区间–左闭右闭写法
这里定义target在左闭右闭区间–[low, high]
- while (low <= high) // 边界条件
- if (target < nums [mid]) 也就是目标值一定在 [low, mid-1] 区间里。此时high赋值为mid-1 。
2.2 二分法区间–左闭右开写法
这里定义target在左闭右开区间–[low, high)
即在该区间中不会去对high位置的元素做比较。
- while (low < high) // 边界条件
- if (target < nums [mid]) 也就是目标值一定在 [low, mid) 区间里。此时high赋值为mid 。
- 注意:初始化的high应该等于数组长度。(因为是右开,取不到下标为数组长度的数组元素)
3 代码(均使用左闭右闭)
3.1 Java版本
class Solution {
public int search(int[] nums, int target) {
// 避免当 target 小于nums[0] nums[nums.length - 1]时多次循环运算
if (target < nums[0] || target > nums[nums.length - 1]) {
return -1;
}
int low = 0, high = nums.length - 1;
int mid = low + ((high - low) >> 1);
while (low <= high) {
if (nums[mid] == target)
return mid;
else if (nums[mid] < target)
low = mid + 1;
else if (nums[mid] > target)
high = mid - 1;
mid = low + ((high - low) >> 1);
}
return -1;
}
}
注意:mid = low + ((high - low) >> 1)
为什么要这样写呢?
因为,如果按平时的写法mid = (low + high) / 2 ,这里的low,high,mid都是int类型,想象这样一种情况,如果high足够大,大到low+high超出了int的表示范围,那会造成出错。
所以这里使用了这样的写法 >>1 是指在二进制数中,右移一位,也就是除以2的意思。而且位运算比除法运算快的多。
3.2 C++版本
class Solution {
public:
int search(vector<int>& nums, int target) {
int left = 0;
int right = nums.size() - 1; // 定义target在左闭右闭的区间里,[left, right]
while (left <= right) { // 当left==right,区间[left, right]依然有效,所以用 <=
int middle = left + ((right - left) / 2);// 防止溢出 等同于(left + right)/2
if (nums[middle] > target) {
right = middle - 1; // target 在左区间,所以[left, middle - 1]
} else if (nums[middle] < target) {
left = middle + 1; // target 在右区间,所以[middle + 1, right]
} else { // nums[middle] == target
return middle; // 数组中找到目标值,直接返回下标
}
}
// 未找到目标值
return -1;
}
};
3.3 C版本
int search(int* nums, int numsSize, int target){
int left = 0;
int right = numsSize-1;
int middle = 0;
//若left小于等于right,说明区间中元素不为0
while(left<=right) {
//更新查找下标middle的值
middle = (left+right)/2;
//此时target可能会在[left,middle-1]区间中
if(nums[middle] > target) {
right = middle-1;
}
//此时target可能会在[middle+1,right]区间中
else if(nums[middle] < target) {
left = middle+1;
}
//当前下标元素等于target值时,返回middle
else if(nums[middle] == target){
return middle;
}
}
//若未找到target元素,返回-1
return -1;
}
3.4 JavaScript版本
/**
* @param {number[]} nums
* @param {number} target
* @return {number}
*/
var search = function(nums, target) {
// right是数组最后一个数的下标,num[right]在查找范围内,是左闭右闭区间
let mid, left = 0, right = nums.length - 1;
// 当left=right时,由于nums[right]在查找范围内,所以要包括此情况
while (left <= right) {
// 位运算 + 防止大数溢出
mid = left + ((right - left) >> 1);
// 如果中间数大于目标值,要把中间数排除查找范围,所以右边界更新为mid-1;如果右边界更新为mid,那中间数还在下次查找范围内
if (nums[mid] > target) {
right = mid - 1; // 去左面闭区间寻找
} else if (nums[mid] < target) {
left = mid + 1; // 去右面闭区间寻找
} else {
return mid;
}
}
return -1;
};
4 小结
注意再循环中坚持循环不变量规则。也就是要么坚持左闭右闭,要么坚持左闭右开,边界条件和函数体内的赋值要遵循同样的原则。
本篇博客只记录了一些博主需要的知识点。如果还有不理解的地方可以点击链接跳转至Carl老师的代码随想录
里面有更多语言版本和图解。
By–Suki 2022/12/30