原题地址:https://oj.leetcode.com/problems/search-insert-position/;
查找插入口,在使用二分查找发的时候需要注意,与以前不同。在没有找到与target相同的元素的时候,不能回复“找不到”,而应该更具情况返回。边界情况是target小雨第一个元素,或target大于最后一个元素。如果计算出的mid位置上的元素刚好等于target,则可以直接返回mid。如果循环因会还条件越界而终止,则返回最后一个beg。(annotation见代码)
还需要注意的是,beg和end在每一次更新的时候不能都用mid开赋值,这样会造成死循环。因为,当beg == end - 1的时候,如果用beg = mid来更新的话,则等于没有更新因为此时 beg + (end - beg) / 2 == beg,其中(end - beg) / 2 == 0。
public class Solution {
public int searchInsert(int[] A, int target) {
int beg = 0, end = A.length - 1, mid, res;
if(target == A[end]) return end;
if(target > A[end]) return end + 1;
if(target <= A[beg]) return beg;
while(beg < end)
{
mid = beg + (end - beg) / 2;
if(A[mid] < target)
beg = mid + 1;
else if(A[mid] == target)
return mid;
else
end = mid; //when end pass through end, the loop will stop.
}
res = beg;
return res;
}
}