LintCode : 搜索旋转排列数组

题目

假设有一个排序的按未知的旋转轴旋转的数组(比如，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

思路

首先用二分法确定mid的位置，判断A[mid]与A[r], A[p]的关系，找出mid的左边或者右边为顺序排列的，如果A[mid] > A[r]则mid左边为顺序排列，else则mid右边顺序排列，再判断target从属于哪一边，这样循环的查找。

代码

int search(vector<int> &A, int target) {        long long p = 0;         long long r = A.size() - 1;        while(p <= r) {            long long q = (p + r) / 2;            if(A[q] == target) {                return q;            }            else if(A[q] > A[r]) {                if(target <= A[r] || target > A[q]) {                    p = q + 1;                }                else {                    r = q - 1;                }            }            else {                if(target > A[q] && target <= A[r]) {                    p = q + 1;                }                else {                    r = q - 1;                }            }        }        return -1;    }