LeetCode 33: Search in Rotated Sorted Array

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.

解题思路

使用二分搜索（Binary Search）进行查找。因为nums[] 只是一个 rotated sorted array，在判断是往前半部分还是后半部分搜索时不能直接根据 target 和 nums[middle] 的大小来判断了。又因为 nums[] 中没有重复的元素，所以以 middle 为界划分得到的两部分子数组中至少有一部分是严格递增的，如下图所示：

我们可以很容易地判断 target 是否在这递增的子数组中，从而可以判断是往前半部分还是后半部分搜索。

代码如下：

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

        while(left <= right) {
            int mid = (left + right) >> 1;

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

            if (nums[left] <= nums[mid]) { // 左半部分有序
                if (nums[left] <= target && target < nums[mid]) { 
                    // target 在左半部分（如存在的话）
                    right = mid - 1;
                }
                else { 
                    // target 在右半部分（如存在的话）
                    left = mid + 1;
                }
            }
            else { // 右半部分有序
                if (nums[mid] < target && target <= nums[right]) {
                    // target 在右半部分（如存在的话）
                    left = mid + 1;
                }
                else {
                    // target 在左半部分（如存在的话）
                    right = mid - 1;
                }
            }
        }

        return -1;
    }
};