原题链接: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; } };