33. 搜索旋转排序数组

题目描述

假设按照升序排序的数组在预先未知的某个点上进行了旋转。( 例如，数组 [0,1,2,4,5,6,7] 可能变为 [4,5,6,7,0,1,2] )。搜索一个给定的目标值，如果数组中存在这个目标值，则返回它的索引，否则返回 -1 。

你可以假设数组中不存在重复的元素。你的算法时间复杂度必须是 O(log n) 级别。

示例 1:

输入: nums = [ 4, 5, 6, 7, 0, 1, 2 ], target = 0 。输出: 4

示例 2:

输入: nums = [4,5,6,7,0,1,2], target = 3 。输出: -1

 

思路1

双指针。但是这种时间复杂度达不到log(n)。

 

代码1

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int len = nums.size();
        if (len < 1)
            return -1;
        if (len == 1){
            if (nums[0] == target)
                return 0;
            else
                return -1;
        }
        int i = 0, j = len - 1;
        while (i <= j){
            if (nums[j] == target)
                return j;
            if (nums[i] == target)
                return i;
            if (nums[j] > nums[i]){
                if (nums[j] < target)
                    return -1;
            }
            j--;
            i++;
        }
        return -1;
    }
};

 

思路2

由于时间复杂度要求为log(n)，所以很好就想到二分查找。但是这个二分查找需要分情况考虑：

（1）若nums[mid] == target，则直接返回

（2）若nums[0] <= nums[mid]，则此时表明（0~mid不包含旋转），这时若nums[0] <= target < nums[mid]，则在左边区域搜索；否则在右边区域搜索；（这里为什么要nums[mid] >= nums[0]，因为要考虑左边只有一个数的情况）

（3）若nums[0] > nums[mid]，这个时候若nums[mid] > target 或者 target >= nums[0]，则说明target在左边区域，否则在右边区域。

这种题自己整几个特殊例子就能找到规律了。如[1,3]找3，[3 4 5 6 7 1 2]找7等。

 

代码2

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int n = nums.size();
        if (n == 0) return -1;
        int left = 0, right = n-1;
        while (left <= right) {
            int mid = (left+right) / 2;
            if (nums[mid] == target) return mid;
            if (nums[mid] >= nums[0]){
                if (nums[mid] > target && nums[0] <= target) right = mid-1;
                else left = mid+1;
            }
            else {
                if (target >= nums[0] || nums[mid] > target) right = mid-1;
                else left = mid+1;
            }
        }
        return -1;
    }
};