[Leetcode] 35 - Search Insert Position

原题链接:https://oj.leetcode.com/problems/search-insert-position/


这道题呢,其实也是二分查找的变种。思路如下,

1. 因为涉及到插入位置,那么首先考虑边界问题,0或者n号位置。

2. 刚才提到二分查找,那么首先我们每次循环有left,right,mid位置。如果A[mid] >= target而且A[mid - 1] < target,那么mid则是应该需要插入的位置。这里就要考虑到第一点的边界问题,如果插入位置在最前或者最后,那应该怎么变换。

3. 我们将数组从0 ~ n - 1扩大到0 ~ n,对于最前和最后进行补值,最前面为INT_MIN,最后为INT_MAX。这样再进行二分查找就覆盖了所有的条件。


class Solution {
public:
    int searchInsert(int A[], int n, int target) {
        if (n <= 0) return 0;
        
        int left = 0;
        int right = n;
        
        int A_1 = INT_MIN;
        int A_N_1 = INT_MAX;
        
        while (left <= right) {
            int mid = (left + right) / 2.0;
            
            int A_mid = mid < n ? A[mid] : INT_MAX;
            int A_mid_1 = mid >= 1 ? A[mid - 1] : INT_MIN;
            
            if (target <= A_mid) {
                if (target > A_mid_1) {
                    return mid;
                } else {
                    right = mid - 1;
                }
            } else {
                left = mid + 1;
            }
        }
        
        return left;
    }
};


你可能感兴趣的:([Leetcode] 35 - Search Insert Position)