Lintcode 搜索旋转排序数组

假设有一个排序的按未知的旋转轴旋转的数组(比如，0 1 2 4 5 6 7 可能成为4 5 6 7 0 1 2)。给定一个目标值进行搜索，如果在数组中找到目标值返回数组中的索引位置，否则返回-1。

你可以假设数组中不存在重复的元素。

您在真实的面试中是否遇到过这个题？ 

Yes

样例

给出[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;
        
    }
};

如有问题请留言。

转载请注明出处。