题目描述
给定一个排序数组和一个目标值,在数组中找到目标值,并返回其索引。如果目标值不存在于数组中,返回它将会被按顺序插入的位置。
你可以假设数组中无重复元素。
示例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 }