[leetcode] 33. 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.

这道题是在前后两段颠倒后的有序数组中查找给定值，题目难度为Hard。

题目可以参考二分查找法，只是这里数组被颠倒过了，所以对中间元素的比对更加复杂。这里分别用l、m、r表示二分查找过程中左端、中间和右端的元素下标。

• 如果target==nums[m]，则查找结束。

• 如果target>nums[m]，此时分为三种情况。

  • 如果target>nums[l]，需比较nums[m]和nums[l]：若nums[m]>=nums[l]，表明目标值在后半段，所以将l置为m+1；若nums[m]<nums[l]，表明目标值在前半段，所以将r置为m-1。

  • 如果target<nums[l]，此时只有一种情况，目标值在后半段，将l置为m+1。

  • 如果target==nums[l]，则查找结束。
    

• 如果target<nums[m]，此时也分为三种情况。

  • 如果target<nums[l]，需比较nums[m]和nums[l]：若nums[m]>=nums[l]，表明目标值在后半段，所以将l置为m+1；若nums[m]<nums[l]，表明目标值在前半段，所以将r置为m-1。

  • 如果target>nums[l]，此时只有一种情况，目标值在前半段，将r置为m-1。

  • 如果target==nums[l]，则查找结束。

• 如果循环结束没有找到，则返回-1。

整个判断过程还是比较绕的，也不一定用nums[l]、nums[m]和target三者进行比较，整个判断过程其实就是将三者依大小进行排列组合，根据各种情况进行二分查找。数组如果未颠倒也是可以查找的。具体代码：

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