搜索插入位置 - 码农教程

问题描述：

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

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

示例 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解决方案

简单的二分查找，找到不小于给定元素的第一个元素位置。

class Solution {
    // 找到第一个大于等于当前元素值的位置
    public int searchInsert(int[] nums, int target) {
        int left = 0;
        int right = nums.length - 1;
        if(target > nums[right]){
            return nums.length;
        }
        while(left < right - 1){
            int mid = left + (right - left) / 2;
            if(nums[mid] > target){
                right = mid;
            }else if(nums[mid] < target){
                left = mid;
            }else{
                return mid;
            }
        }
        if(nums[left] >= target){
            return left;
        }
        return right;
    }
}

时间复杂度为O(Nlog(N)),其中N为数组长度。