剑指 Offer II 068. 查找插入位置
给定一个排序的整数数组 nums 和一个整数目标值 target ,请在数组中找到 target ,并返回其下标。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
请必须使用时间复杂度为 O(log n) 的算法。
输入: nums = [1,3,5,6], target = 5
输出: 2
题目有要求,要求时间复杂度为O(log n),所以采用二分查找,首先是如果能找到对应的元素,那么就直接返回下标即可,如果找不到,最后就直接返回left,left就是要插入元素的下标应该所对应的位置。
分析:【1,3,5,6】,tarfet=2
执行过程:
left=0,right=3,mid=2, 5>2,right=1
left=0,right=1,mid=1, 3>2,right=0
left=0,right=0,mid=0, 0>2,left=1
前提是在while循环里面必须通过的条件就是left可以等于right,这样最后一次运算就可以得到位置。
在这里补充一个方法:binarySearch()方法。
查找指定元素在数组里面的位置:
如果要找的元素在数组里面,就返回索引
如果不存在:
(1)在数组元素范围之内,返回-插入点索引值
(2)小于数组范围,返回-1
(3)大于数组范围,返回-(length+1)
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){
right=mid-1;
}
else if(nums[mid]<target){
left=mid+1;
}
else if(nums[mid]==target){
return mid;
}
}
return left;
}
}