35. 搜索插入位置

题目描述

给定一个排序数组和一个目标值，在数组中找到目标值，并返回其索引。如果目标值不存在于数组中，返回它将会被按顺序插入的位置。

你可以假设数组中无重复元素。

示例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
}