704.二分查找
题目的前提
有序数组、数组中无重复元素。不重复是使用二分法重点前提之一。
题目考察的难点(边界条件)
- 结束循环的条件
- 循环中right值的更新,是right=middle还是right=middle-1
区间的定义就是不变量(循环不变量),这个是需要保持的。
两种写法
区间的定义一般为两种,左闭右闭即[left, right],或者左闭右开即[left, right)。
左始终是闭的,右边不确定。
左闭右闭[left, right]
- while (left <= right) 要使用 <= ,因为left == right是有意义的,所以使用 <=
- if (nums[middle] > target) right 要赋值为 middle - 1,因为当前这个nums[middle]一定不是target,那么接下来要查找的左区间结束下标位置就是 middle - 1
- 时间复杂度:O(log n),空间复杂度:O(1)
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 left = 0, right = nums.length - 1;
while (left <= right) {
int mid = left + ((right - left) >> 1);
if (nums[mid] == target)
return mid;
else if (nums[mid] < target)
left = mid + 1;
else if (nums[mid] > target)
right = mid - 1;
}
return -1;
}
}
为什么使用left+((right - left))>> 1而不使用(left + right)>> 1
是为了防止int类型数据溢出,因为left+right可能是一个比32bit大的数(int 最大值为 2147483647),先求差再加left可以防止溢出。
左闭右开[left, right)
- while (left < right),这里使用 < ,因为left == right在区间[left, right)是没有意义的
- if (nums[middle] > target) right 更新为 middle,因为当前nums[middle]不等于target,去左区间继续寻找,而寻找区间是左闭右开区间,所以right更新为middle,即:下一个查询区间不会去比较nums[middle]
- 时间复杂度:O(log n),空间复杂度:O(1)
class Solution {
public int search(int[] nums, int target) {
int left = 0, right = nums.length;
while (left < right) {
int mid = left + ((right - left) >> 1);
if (nums[mid] == target)
return mid;
else if (nums[mid] < target)
left = mid + 1;
else if (nums[mid] > target)
right = mid;
}
return -1;
}
}
参考资料:704. 二分查找