leetcode 35 搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
示例 1:
输入: nums = [1,3,5,6], target = 5
输出: 2
示例 2:
输入: nums = [1,3,5,6], target = 2
输出: 1
示例 3:
输入: nums = [1,3,5,6], target = 7
输出: 4
示例 4:
输入: nums = [1,3,5,6], target = 0
输出: 0
示例 5:
输入: nums = [1], target = 0
输出: 0
二分法一
public int searchInsert(int[] nums, int target) {
int left=0;
int right=nums.length-1;
//索引范围[left,right],但是如果while循环条件是left<right,那么left==right计算出的mid没有遍历到,所以循环条件必须要加等于
while(left<=right){
int mid=left+((right-left)>>1);
if(nums[mid]==target){
return mid;
}else if(nums[mid]>target){
//right=mid-1,而不是right=mid,因为一开始索引取值范围是
//[left,right],这边要和上面保持一致
right=mid-1;
}else{
left=mid+1;
}
}
//while退出条件是left>right即left=right+1,能走到这,说明没有找到
// target,我们要返回target的插入位置,因为right索引对应的值肯定小于
//target,比如target是5,right索引对应的值为4,最后返回的肯定是
//right+1即left。这边大家可以尝试带入几个数字试着模拟一下就明白了。
return left;//这边只能返回left,不能返回right
}
二分法二
public int searchInsert(int[] nums, int target) {
int left=0;
int right=nums.length;
//索引范围[left,right),当left==right退出循环。如果写成<=那么
//left==right会造成下面的数组角标越界,这边循环条件只能写<。
while(left<right){
int mid=left+((right-left)>>1);
if(nums[mid]==target){
return mid;
}else if(nums[mid]>target){
right=mid;//为了和上面索引范围保持一致即[left,right),不能写成mid-1;
}else{
left=mid+1;
}
}
return left;//返回right也可以
}