leetcode 刷题之路 26 Search in Rotated Sorted Array

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.

题目要求从一个旋转过得有序数组查找一个给定数字。

采用遍历方法可以完成查找但是时间复杂度为O(n)，会引起超时，而且并没有利用数组旋转前有序这个性质。

这道题可以采用二分查找法解决，但是需要根据数组旋转的特征做出调整。有序数组旋转后内部可能会出现两个升序序列，并且前一个升序序列必然大于后一个升序序列。因此我们在进行二分查找的时候首先需要根据A[mid]和A[begin]的值大小确定mid位于哪个序列当中，然后再根据target的值做出具体的判断。更具体的逻辑可以参考程序。

测试通过代码：

class Solution {
public:
	int search(int A[], int n, int target)
	{
		int begin = 0, end = n - 1,mid;
		while (begin <= end)
		{
			mid = (begin + end) / 2;
			if (A[mid] == target)
				return mid;
			if (A[mid]>=A[begin])
			{
				if (target >= A[begin] && target < A[mid])
					end = mid - 1;
				else
					begin = mid + 1;
			}
			else
			{
				if (target>A[mid] && target <= A[end])
					begin = mid + 1;
				else
					end = mid - 1;
			}
		}
		return -1;
	}
};

更新，和Search in Rotated Sorted Array II 题目统一，增加重复数字的判断。

accepted answer:

class Solution {
public:
	int search(int A[], int n, int target)
	{
		int mid;
        int begin=0,end=n-1;
        while(begin<=end)
        {
            mid=(begin+end)/2;
            if(A[mid]==target)
                return mid;
            if(A[mid]>A[begin]) //两个升序序列的话，mid必然位于第一个序列，因为第二个序列的数都小于等于begin
            {
                if(target>=A[begin]&&target<A[mid])
                    end=mid-1;
                else
                    begin=mid+1;
            }
            else if(A[mid]<A[begin])//两个升序序列的话，mid必然位于第二个序列，因为第一个序列的数都大于等于begin
            {
                if(target>A[mid]&&target<=A[end])
                    begin=mid+1;
                else
                    end=mid-1;
            }
            else                //相等的时候，说明begin位置数字有重复，重复元素位于第一个序列的前面，或者第二个序列后面
            {
                if(A[mid]==A[end])  //此条件下，不能判断mid位于哪个位置的重复元素中，只能移动最小的位置
                {
                    ++begin;
                    --end;
                }
                else                //此条件下，重复数字都位于第一个序列，故begin和mid之间所有数字都相同
                    begin=mid+1;
            }
        }
        return -1;
	}
};