35. Search Insert Position

刷题笔记

35. Search Insert Position

题目

Given a sorted array of distinct integers 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.

思路

1. 我的思路比较繁琐，首先是判断如果target大于最大的，就直接返回最大坐标+1；如果target小于最小的，就直接返回0.
2. 进入binary search的判断，我的思路是让最后的情况缩小到两个相邻的元素，如果有匹配就直接返回mid；如果没有匹配，就是target两侧的两个元素，那么这时候返回right，就是要插入的坐标。

代码实现

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

再来看看most vote的解法：

public int searchInsert(int[] A, int target) {
        int low = 0, high = A.length-1;
        while(low<=high){
            int mid = (low+high)/2;
            if(A[mid] == target) return mid;
            else if(A[mid] > target) high = mid-1;
            else low = mid+1;
        }
        return low;
    }

这种解法的思路更精巧一些：

1. 首先范围收缩到一个元素，这时候有这样几种情况。
2. 第一种情况是收缩到一个元素之前，mid匹配，返回mid值。
3. 第二种情况是收缩到一个元素，然后mid就是这个元素，再次进入循环，如果这个元素比target大，则让high-1，跳出循环，low还在原位置，返回low，low正好是target的插入下标；如果这个元素比target小，则让low + 1，跳出循环，high还在原位置，而low已经到最后收缩位置的下一个位置，即正好也是插入位置，此时返回low也正好是正确插入位置。