给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例 1:
输入: [1,3,5,6], 5
输出: 2
示例 2:
输入: [1,3,5,6], 2
输出: 1
示例 3:
输入: [1,3,5,6], 7
输出: 4
示例 4:
输入: [1,3,5,6], 0
输出: 0
//我写的破代码,本来第一时间想起用二分查找,但是发现无从下手,故直接遍历
class Solution {
public int searchInsert(int[] nums, int target) {
for(int i=0;i<nums.length;i++){
if(target<=nums[i]){
return i;
}
}
return nums.length;
}
}
//下面给出一个二分查找的代码示例
//二分查找模板
class Solution {
public int searchInsert(int[] nums, int target) {
int left = 0, right = nums.length - 1; // 注意
while(left <= right) { // 注意
int mid = (left + right) / 2; // 注意
if(nums[mid] == target) { // 注意
// 相关逻辑
} else if(nums[mid] < target) {
left = mid + 1; // 注意
} else {
right = mid - 1; // 注意
}
}
// 相关返回值
return 0;
}
}
正解
class Solution {
public int searchInsert(int[] nums, int target) {
int left = 0, right = nums.length - 1;
while(left <= right) {
int mid = (left + right) / 2;
//还可以有下面的写法---比较推荐
//int mid = left + (right - left) / 2;
//int mid = ((right - left) >> 1) + left;
//int mid = (left + right) >>> 1;
if(nums[mid] == target) {
return mid;
} else if(nums[mid] < target) {
left = mid + 1;
} else {
right = mid - 1;
}
}
return left;//注意:返回的是left
}
}
在这里补充一个小细节,我只在写 Java 代码的时候这样用。
int mid = left + (right - left) / 2;
还可以写成:
int mid = (left + right) >>> 1;
这是参考 Java 的 JDK 中 Arrays.binarySearch()
函数的写法。理由是 left + right
即使是在整型溢出以后,由于无符号右移 >>> 1
,仍然能够得到正确的结果(我掌握的语言中,只有 Java 语言中有 >>>
这个操作符)。
虽然 >> 1
和 /2
,但是有些语言编译器都会将 /2
转换成位运算的操作,这是编译器内部的优化。因此我们没有必要手动去做这一步优化,写代码的时候还是写 /2
。