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