题目描述
有一个排过序的数组,包含n个整数,但是这个数组向左进行了一定长度的移位,例如,原数组为[1,2,3,4,5,6],向左移位5个位置即变成了[6,1,2,3,4,5],现在对于移位后的数组,需要查找某个元素的位置。请设计一个复杂度为log级别的算法完成这个任务。
给定一个int数组A,为移位后的数组,同时给定数组大小n和需要查找的元素的值x,请返回x的位置(位置从零开始)。保证数组中元素互异。
测试样例:
[6,1,2,3,4,5],6,6
返回:0
思路:使用二分法,不过要多加些条件。
代码:
class Finder {
public:
int findElement(vector<int> A, int n, int x) {
// write code here
int i=0;
int j=n-1;
int mid;
while(i<=j){
mid=(i+j)/2;
if(A[mid]==x)
return mid;
else if(x>A[mid]){
if(A[i]>A[mid]&&x>A[j])//第一个条件防止数组没有旋转的情况
j=mid-1;
else
i=mid+1;
}
else{
if(A[i]<A[mid]&&x<A[i])//第一个条件防止数组没有旋转的情况
i=mid+1;
else
j=mid-1;
}
}
return -1;
}
};