刷题小记
力扣35题-搜索插入位置
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
这题是在二分法基础上完成的!
第一种情况:刚好找到,返回下标即可!
第二种情况:不存在,
1,target比nums[mid]要小,此时插入mid元素之前,mid后移一位,所以新元素下标为mid。
2,target比nums[mid]要大,此时插入mid元素之后,mid不动,所以新元素下标为mid+1。
代码如下!
class Solution {
public int searchInsert(int[] nums, int target) {
int low=0;
int high=nums.length-1;
int mid=0;
while(low<=high){
mid=(low+high)/2;
if(target>nums[mid]){
low=mid+1;
}else if(target<nums[mid]){
high=mid-1;
}else{
break;
}
}
if(target==nums[mid]) return mid;
else if(target>nums[mid]) return mid+1;
else if(target<nums[mid]) return mid;
return mid;
}
}