[Leetcode] Search Insert Position 搜索插入位置

442 查看

Search Insert Position

Given a sorted array and a target value, return the index if the target is found. If not, return the index where it would be if it were inserted in order.

You may assume no duplicates in the array.

Here are few examples.
    [1,3,5,6], 5 → 2
    [1,3,5,6], 2 → 1
    [1,3,5,6], 7 → 4
    [1,3,5,6], 0 → 0

二分搜索法

复杂度

时间 O(logN) 空间 O(1)

思路

这是最典型的二分搜索法了。如果使用我这个二分模板是可以直接适用的,在模板中,while循环的条件是min<=max,如果target和mid指向的值相等,则返回mid,否则根据情况min = mid + 1或者max = mid - 1。这样的好处是,如果找不到该数,max是比该数小的那个数的下标,而min是比该数大的那个数的下标。这题中,我们返回min就行了,如果返回max,要注意-1的情况。

代码

public class Solution {
    public int searchInsert(int[] nums, int target) {
        int min = 0, max = nums.length - 1;
        // 条件是min <= max
        while(min <= max){
            int mid = min + (max - min) / 2;
            // 找到了
            if(nums[mid] == target){
                return mid;
            }
            // 在左边
            if(nums[mid] > target){
                max = mid - 1;
            } else {
            // 在右边
                min = mid + 1;
            }
        }
        return min;
    }
}