LeetCode 34. Search Insert Position

二分查找插入的位置。

若恰好匹配num[mid] == target, 返回mid;

若num[mid] < target && num[mid+1] > target, 返回mid+1.

注意边界检查。

代码：

class Solution {public:    int searchInsert(int A[], int n, int target)     {    int l=0, r=n-1, mid;    while (l <= r)    {    mid = (l+r) >> 1;    if (A[mid] == target)    {    return mid;    } else if (A[mid]<target && (mid+1==n || A[mid+1]>target)) // 边界检查    {    return mid+1;    } else if (A[mid] < target)    {    l = mid + 1;    } else    {    r = mid - 1;    }    }        }};