之前第一次写的方法是O(n)的,回来再看,发现很明显是一道O(logN) 的二分查找题目,果断重新写。要判断target在array中的位置或者插入位置,那么就A[mid]和target 比较,发现A[mid] == target则返回mid,当start 和end 相遇的时候跳出循环。这时候有三种情况,找到了,那么return start = end都一样; A[start] 比较大,说明target要被插入到start; target比较大,那么target 要插在start + 1的地方。 前面两种情况可以写成一条判断,后面一种写成一种。
public int searchInsert(int[] A, int target) {
if(A == null) {return 0;}
int start = 0;
int end = A.length - 1;
int mid = (start + end) / 2;
while(start < end) {
mid = (start + end) /2;
if(A[mid] == target) return mid;
if(A[mid] > target) end = mid;
if(A[mid] < target) start = mid + 1;
}
if(A[start] >= target) return start;
else return start + 1;
}