[LeetCode] Search in Rotated Sorted Array

题目

Suppose an array sorted in ascending order 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.

你将获得一个目标值进行搜索。 如果在数组中找到返回其索引，否则返回-1。

You may assume no duplicate exists in the array.

你可以假定数组中不存在重复。

分析

其实这就是一个循环有序数组，循环有序数组就是将一个有序数组切成两段，并交换位置。

主要思路还是利用二分查找，但由于循环的原因，划分数组时会出现不同情况，因此要做进一步的判断。

1. 当目标元素 = 中间元素时，直接返回中间元素索引。
2. 当中间元素 < 起始元素时，从中间元素到结束元素是有序的。此时可以判断目标元素是否在有序部分中。
   若目标元素在有序部分中，在其中进行二分查找。否则在循环部分中进行查找。
3. 当中间元素 > 起始元素时，从起始元素到中间元素时有序的。此时可以判断目标元素是否在有序部分中。
   若目标元素在有序部分中，在其中进行二分查找。否则在循环部分中进行查找。

注：由于本题是判断目标元素的位置，和在循环有序数组找最小值可以直接认定循环又有序部分的起始元素就是最小值不同，需要一步步二分查找，故循环条件是起始元素 ≤ 结束元素，而不是起始元素小于结束元素即可。

代码

#include <iostream>
#include <vector>

using namespace std;

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int start = 0, end = nums.size() - 1;

        while (start <= end) {
            int mid = (start + end) / 2;

            if (target == nums[mid]) {
                return mid;
            }

            if (nums[mid] < nums[start]) {
                if (target > nums[mid] && target <= nums[end]) {
                    start = mid + 1;
                } else {
                    end = mid - 1;
                }
            } else {
                if (target >= nums[start] && target < nums[mid]) {
                    end = mid - 1;
                } else {
                    start = mid + 1;
                }
            }
        }
        return -1;
    }
};

int main() {
    int num[] = {7, 8, 0, 1, 2, 3, 4, 5, 6};
    int target = 0;
    int length = sizeof(num) / sizeof(int);
    vector<int> nums(num, num + length);

    Solution solution;
    int index = solution.search(nums, target);
    cout << "The index is " << index << endl;
}

参考资料

CSDN 博客：http://blog.csdn.net/linhuanmars/article/details/20525681