首先开始最基本的Binary Search, 数组是有序的,但是有重复数。
例题: Search for a Range
复杂度:时间O(log n), 空间O(1).
分析: 首先需要复杂度在O(log n)左右,显然暗示了Binary Search是首选,然后注意这道题是需要找到一个range,并不是找到target的位置就可以了。
class Solution{
public:
vector searchRange(int A[], int n, int target) {
vector result(2,-1);
int index = search(A, n, target);
if(index == -1) return result;
int l = index, r = index;
while(l -1 >= 0 && A[l] == A[l-1]) --l;
while(r +1 <= n-1 && A[r] == A[r+1]) ++r;
result[0] = l;
result[1] = r;
return result;
}
int search(int A[], int n , int target){
if(A == nullptr || n == 0) return -1;
int left = 0, right = n -1;
while(left <= right){
int mid = left + (right - left) /2;
if(target > A[mid])
left = mid + 1;
else if(target < A[mid])
right = mid - 1;
else if(target == A[mid])
return mid;
}
return -1;
}
};
注意这里做binary search时候,对于A[mid]分为了三种不同情况讨论,相等,小于和大于,并且while循环的终止条件是left <= right,也就是说最后left > right时候表明目标数肯定不存在了。在对于不同的问题,这些边界条件会相应的改变。另外这样搜索如果最后目标数存在,最后结果一定是以A[mid]的形式。
例题: Search Insert Position
复杂度:时间O(log n), 空间O(1).
分析:这道题的关键是如果目标数不存在,必须返回目标数可以插入的位置,这样经典的binary search返回A[mid]的方法(如上题)没法直接实现因此必须对循环的条件以及边界进行调整。首先当循环结束时候,为了能返回一个目标数可以插入的地址,我们必须让left == right也就是说循环条件相应调整为left < right. 如果需要设置循环里面的边界条件和left, right的初始条件,我们必须看一下最后一次循环时候的情况:最后一次循环时, left + 1 = right只剩下两个元素需要考虑(这里忽略数组只有一个元素的情形,因为一个元素时,程序根本不会进入循环),此时mid = left,如果target > A[mid], left = mid + 1 = right与之前情况类似,那么当target < A[mid]时候也就是说target并不存在,此时应该输出target元素需要插入的位置也就是left, 因此right = mid = left,当target = A[mid]情形与小于相同,right = mid = left.还有一种情况,当target大于数组里面所有数的时候,插入位置应该在所有数组之后,因此right的初始值必须是n,因为很容易发现前面的边界条件设置导致最后输出的left都会在两个初始值left和right之间。
本文来自电脑杂谈,转载请注明本文网址:
http://www.pc-fly.com/a/jisuanjixue/article-26574-1.html