35、搜索插入位置

搜索插入位置

题目描述

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

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

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

思路分析

基于二分法的思想查找待插入的位置。代码如下：

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