LeetCode 35.搜索插入位置

1.题目描述

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

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

示例 1:

输入: [1,3,5,6], 5
输出: 2
示例 2:

输入: [1,3,5,6], 2
输出: 1
示例 3:

输入: [1,3,5,6], 7
输出: 4
示例 4:

输入: [1,3,5,6], 0
输出: 0

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/search-insert-position
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

2.思路

2.1 从头遍历

可以对nums数组从前向后遍历，查找target的出现位置或插入位置。
下面给出该方法的一个比较简单的实现代码：

class Solution {
    public int searchInsert(int[] nums, int target) {
        for(int i = 0; i < nums.length; i++){
            if(nums[i] >= target){
                return i;
            }
        }
        return nums.length;
    }
}

可以看到，只需要在找到第一个不小于target的nums数组中的元素即可。

2.2 二分法查找

由于数组已经排好序，所以用二分法查找效率会更高。而查找的思路与 2.1 一样，都是去查找第一个不小于target的nums数组中的元素即可。
代码如下：

public int searchInsert(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        int result = nums.length;
        while(left <= right){
            int mid = (right + left) / 2;
            if(target <= nums[mid]){
                result = mid;
                right = mid - 1;
            }else{
                left = mid + 1;
            }
        }
        return result;
    }