Suppose a sorted array is rotated at some pivot unknown to you beforehand.
(i.e., 0 1 2 4 5 6 7
might become 4 5 6 7 0 1 2
).
You are given a target value to search. If found in the array return its index, otherwise return -1.
You may assume no duplicate exists in the array.
这道题是二分查找Search Insert Position的变体,看似有点麻烦,其实理清一下还是比较简单的。因为rotate的缘故,当我们切取一半的时候可能会出现误区,所以我们要做进一步的判断。具体来说,假设数组是A,每次左边缘为start,右边缘为end,还有中间位置是mid。在每次迭代中,分三种情况:
(1)如果target==A[mid],那么mid就是我们要的结果,直接返回;
(2)如果A[mid]>=A[start],那么说明从start到mid一定是有序的(没有受到rotate的影响),那么我们只需要判断target是不是在start到mid之间,如果是则把右边缘移到m-1,否则就target在另一半,即把左边缘移到m+1。
(3)如果A[m]<A[start],那么说明从mid到end一定是有序的,同样只需要判断target是否在这个范围内,相应的移动边缘即可。
根据以上方法,每次我们都可以切掉一半的数据,所以算法的时间复杂度是O(logn),空间复杂度是O(1)。代码如下:
class Solution {
public:
int search(int A[], int n, int target) {
int start = 0;
int end = n-1;
while(start<= end) {
int mid = (start+end)/2;
if(A[mid] == target)
return mid;
if(A[mid]>=A[start]) {
if(A[start]<= target && A[mid]> target) {
end = mid-1;
} else {
start = mid +1;
}
} else {
if(A[mid]< target && A[end]>= target) {
start = mid + 1;
} else {
end = mid -1;
}
}
}
return -1;
}
};
和Search in Rotated Sorted Array唯一的区别是这道题目中元素会有重复的情况出现。不过正是因为这个条件的出现,出现了比较复杂的case,甚至影响到了算法的时间复杂度。原来我们是依靠中间和边缘元素的大小关系,来判断哪一半是不受rotate影响,仍然有序的。而现在因为重复的出现,如果我们遇到中间和边缘相等的情况,我们就丢失了哪边有序的信息,因为哪边都有可能是有序的结果。假设原数组是{1,2,3,3,3,3,3},那么旋转之后有可能是{3,3,3,3,3,1,2},或者{3,1,2,3,3,3,3},这样的我们判断左边缘和中心的时候都是3,如果我们要寻找1或者2,我们并不知道应该跳向哪一半。解决的办法只能是对边缘移动一步,直到边缘和中间不在相等或者相遇,这就导致了会有不能切去一半的可能。所以最坏情况(比如全部都是一个元素,或者只有一个元素不同于其他元素,而他就在最后一个)就会出现每次移动一步,总共是n步,算法的时间复杂度变成O(n)。代码如下:
class Solution {
public:
bool search(int A[], int n, int target) {
int start = 0;
int end = n-1;
while(start<= end) {
int mid = (start+end)/2;
if(A[mid] == target)
return true;
if(A[mid]>A[start]) {
if(A[start]<= target && A[mid]> target) {
end = mid-1;
} else {
start = mid +1;
}
} else if(A[mid]<A[start]){
if(A[mid]< target && A[end]>= target) {
start = mid + 1;
} else {
end = mid -1;
}
} else {
start++;
}
}
return false;
}
};