题目描述
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例1:
输入: [1,3,5,6], 5 输出: 2
示例2:
输入: [1,3,5,6], 2 输出: 1
思路
1.本题可以使用暴力法直接求解,但时间复杂度为O(n),可以通过二分搜索将其优化为O(logn)。
2.可以直接按照标准的二分搜索编码,只不过将搜索不到的返回值改为左指针就好了。
3.当在数组中搜索不到目标元素时,左指针是刚好大于target的,右指针是刚好小于target的,所以我们将左指针返回便是结果。
Java代码实现
public int searchInsert(int[] nums, int target) {
int left = 0;
int right = nums.length-1;
while(left <= right){
int mid = (left+right)/2;
if(target == nums[mid])
return mid;
else if(target<nums[mid])
right = mid - 1;
else
left = mid + 1;
}
return left;
}Golang代码实现
func searchInsert(nums []int, target int) int {
left,right := 0,len(nums)-1
for left <= right{
mid := (left + right)/2
if nums[mid] == target {
return mid
}else if nums[mid] > target{
right = mid - 1
}else {
left = mid + 1
}
}
return left
}
京公网安备 11010502036488号