假设有一个排序的按未知的旋转轴旋转的数组(比如,0 1 2 4 5 6 7 可能成为4 5 6 7 0 1 2)。给定一个目标值进行搜索,如果在数组中找到目标值返回数组中的索引位置,否则返回-1。
你可以假设数组中不存在重复的元素。
样例
给出[4, 5, 1, 2, 3]和target=1,返回 2
给出[4, 5, 1, 2, 3]和target=0,返回 -1
挑战
O(logN) time
分析:通过分析可以将数组分为两部分分别进行二分法查找,但是这题的难点在于target在前一段还是后一段,这里我们可以注意一个细节,如果数组是旋转过的则A[0]>A[n-1],所以我们可以判断A[0]和target的大小关系判断target在哪一部分,从而在判断A[mid]的时候添加限制条件。
class Solution {
/**
* param A : an integer ratated sorted array
* param target : an integer to be searched
* return : an integer
*/
public:
int search(vector<int> &A, int target) {
// write your code here
if(A.empty())
return -1;
int n=A.size();
int flag=0;
if(A[0]<=target)
flag=1;
else
flag=2;
int first=0,last=n-1,mid;
while(first<last){
mid=(first+last)/2;
if(A[mid]==target)
return mid;
else if(A[mid]<target){
if(flag==1&&A[mid]<A[0])
last=mid-1;
else if(flag==1&&A[mid]>A[0])
first=mid+1;
else if(A[mid]<A[0])
first=mid+1;
else
return -1;
}
else{
if(flag==1)
last=mid-1;
else if(flag==2&&A[mid]<A[0])
last=mid-1;
else
first=mid+1;
}
}
if(A[first]==target)
return first;
else
return -1;
}
};
如有问题请留言。
转载请注明出处。