链接:704.二分查找
给定一个 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(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) { // 当 left 小于等于 right 时继续搜索
int mid = left + ((right - left) >> 1); // 计算中间点
if (nums[mid] == target) { // 如果中间点的值等于目标值
return mid; // 返回中间点的索引
}
else if (nums[mid] < target) { // 如果中间点的值小于目标值
left = mid + 1; // 更新左指针,缩小搜索范围到右半部分
}
else { // 如果中间点的值大于目标值
right = mid - 1; // 更新右指针,缩小搜索范围到左半部分
}
}
// 未找到目标值
return -1;
}
}
需要注意的是>>
符号是一个位运算符,表示右移操作。具体来说,>> 1
会将操作数的二进制表示右移一位,相当于将这个数除以2并向下取整。
int mid = left + ((right - left) >> 1); // 计算中间点
为什么用位移操作
使用 >> 1
计算中间点的原因是它比除法运算快。虽然在现代编译器和处理器中,这个差别已经很小,但使用位移操作仍然是一个习惯用法,特别是在处理大数据时。
具体到这个代码中
left + ((right - left) >> 1)
这行代码做的是在left
和 right
之间找到一个中间点 mid
:
- 计算范围:
right - left
得到当前搜索范围的长度。 - 右移一位:
(right - left) >> 1
相当于(right - left) / 2
,即将范围长度除以2,找到中间点偏移量。 - 加上左边界:
left + ((right - left) >> 1)
表示从左边界left
开始,向右移动偏移量,找到中间点的实际位置。
用法总结
位移操作符 >>
在二分查找中的作用是快速且高效地计算中间点,避免可能的整数溢出(尽管在现代编译器中不太常见)。
如果你更习惯使用除法运算,也可以这样写:
int mid = left + (right - left) / 2;
两种写法功能相同,但使用位移操作是个较好的实践,特别是考虑到一些极端情况下的性能和安全性。