给定一个 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]之间。
#思路
这道题目的前提是数组为有序数组,同时题目还强调数组中无重复元素,因为一旦有重复元素,使用二分查找法返回的元素下标可能不是唯一的,这些都是使用二分法的前提条件,当大家看到题目描述满足如上条件的时候,可要想一想是不是可以用二分法了。
二分查找涉及的很多的边界条件,逻辑比较简单,但就是写不好。例如到底是 while(left < right)
还是 while(left <= right)
,到底是right = middle
呢,还是要right = middle - 1
呢?
大家写二分法经常写乱,主要是因为对区间的定义没有想清楚,区间的定义就是不变量。要在二分查找的过程中,保持不变量,就是在while寻找中每一次边界的处理都要坚持根据区间的定义来操作,这就是循环不变量规则。
写二分法,区间的定义一般为两种,左闭右闭即[left, right],或者左闭右开即[left, right)。
代码如下:(详细注释)
// Main类,程序的入口
class Main {
public static void main(String[] args) {
// 定义一个int数组
int[] nums = {1, 2, 4, 5, 8, 9};
// 要查找的目标数字
int target = 4;
// 创建一个Solution对象
Solution solution = new Solution();
// 调用search方法查找目标数字在数组中的索引
System.out.println(solution.search(nums, target));
}
}
// Solution类,实现二分查找算法
class Solution {
public int search(int[] nums, int target) {
// 如果目标数字小于数组中的最小值或大于数组中的最大值,则无法查找到目标数字,返回-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) / 2);
// 如果中间数字大于目标数字,说明目标数字在左半部分,将右指针移动到mid-1
if (nums[mid] > target){
right = mid -1;
}
// 如果中间数字小于目标数字,说明目标数字在右半部分,将左指针移动到mid+1
else if (nums[mid] < target){
left = mid + 1;
}
// 如果中间数字等于目标数字,说明已找到,返回中间指针
else{
return mid;
}
}
// 如果找不到目标数字,返回-1
return -1;
}
}