33. Search in Rotated Sorted Array (Array;Divide-and-Conquer)

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.

思路: 改动的二分法。有三种情况:正序、右侧rotate、左侧rotate。三种情况分别讨论。

class Solution {
public:
    int search(vector<int>& nums, int target) {
        return binarySearch(nums,0,nums.size()-1, target);
    }
    
    int binarySearch(vector<int>& nums, int start, int end, int target){
        if(start==end){
            if(nums[start]==target) return start;
            else return -1;
        }
        
        int mid = start+ ((end-start)>>1);
        
        //正序
        if(nums[mid]>=nums[start] && nums[mid]<nums[end]){ //mid可能会=start,所以这里要用>=
            if(target <= nums[mid]) return binarySearch(nums,start,mid,target); //mid肯定<end,所以至少舍弃了一个
            else return binarySearch(nums,mid+1,end,target); //mid+1,至少舍弃了一个
        }
        
        //右侧rotate
        else if(nums[mid]>=nums[start] && nums[mid]>nums[end]){
            if(target>=nums[start] && target<=nums[mid]) return binarySearch(nums,start,mid,target);
            else return binarySearch(nums,mid+1,end,target);
        }
        
        //左侧rotate
        else{
            if(target>=nums[start] || target<=nums[mid]) return binarySearch(nums,start,mid,target);
            else return binarySearch(nums,mid+1,end,target);
        }
    }
};

 

posted on 2016-09-16 21:10  joannae  阅读(172)  评论(0编辑  收藏  举报

导航