力扣35
算法学习推荐网站:代码随想录
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置
输入: nums = [1,3,5,6], target = 5
输出: 2
思路:还是使用普通的二分法就能解决掉
解题如下:
方式一:
class Solution {
public int searchInsert(int[] nums, int target) {
int left=0;
int right=nums.length;
while(left<right){
int mid=left+(right-left)/2;
if(nums[mid]==target){
return mid;
}
if(nums[mid]>target){
right=mid;
}
if(nums[mid]<target){
left=mid+1;
}
}
return right;
}
}
很多人可能对前面的代码没有问题,主要就是对没有搜索到结果的情况下为什么返回的是right感到疑惑,注意:我这里写的是right=nums.length,所以left和right相等时是无效的。其实这个很简单,我们如果没有搜索到,最后一次搜索肯定是[a,a+1) ,但是这个时候搜索还是没有搜索到结果,所以需要返回插入的位置了,这个时候是[b,b),所以这个时候已经不会执行while循环了,下一次无论是left等于mid+1还是right=mid,肯定是right==left了,所以想返回插入的位置就直接返回right或者left即可
方式二:
class Solution {
public int searchInsert(int[] nums, int target) {
int left=0;
int right=nums.length-1;
while(left<=right){
int mid=left+(right-left)/2;
if(nums[mid]==target){
return mid;
}
if(nums[mid]>target){
right=mid-1;
}
if(nums[mid]<target){
left=mid+1;
}
}
return right+1;
}
}
注意:我这里写的是right=nums.length-1,所以left和right相等时是有效的,最后一次临界的情况就是[a,a],但是搜索之后a不是我们需要的值,就需要选择插入位置了,下一次无论是left等于mid+1还是right=mid,right肯定都在left的左边了,所以想返回插入的位置就直接返回right即可