Leetcode 33:搜索旋转数组

题目描述

整数数组 nums 按升序排列，数组中的值 互不相同 。

在传递给函数之前，nums 在预先未知的某个下标 k（0 <= k < nums.length）上进行了 旋转，使数组变为 [nums[k], nums[k+1], ..., nums[n-1], nums[0], nums[1], ..., nums[k-1]]（下标 从 0 开始 计数）。例如， [0,1,2,4,5,6,7] 在下标 3 处经旋转后可能变为 [4,5,6,7,0,1,2] 。

给你 旋转后 的数组 nums 和一个整数 target ，如果 nums 中存在这个目标值 target ，则返回它的下标，否则返回 -1 。

 

示例 1：

输入：nums = [4,5,6,7,0,1,2], target = 0
输出：4
示例 2：

输入：nums = [4,5,6,7,0,1,2], target = 3
输出：-1
示例 3：

输入：nums = [1], target = 0
输出：-1

解题思路

这个题目还是用二分法，虽然数组整体可能不是单调的，但是将数组平分之后必然有一半是有序的。
代码如下：

    int search(vector<int>& nums, int target) {
        int n = nums.size();
        if (n == 0) return -1;
        if (n == 1 && nums[0] != target) return -1;
        int left = 0, right = n - 1;
        int mid;
        while (left <= right)
        {
            mid = (left + right) / 2;
            if (target == nums[mid])
            {
                return mid;
            } 
            if (nums[left] <= nums[mid]) // 左边有序
            {
                if (nums[left] <= target && nums[mid] > target) // target在左边
                {
                    right = mid - 1;
                } 
                else 
                {
                    left = mid + 1;
                }
            }
            else // 右边有序
            {
                if (nums[mid] < target && target <= nums[right]) // target在右边
                {
                    left = mid + 1;
                }
                else
                {
                    right = mid - 1;
                }
            }
        }

        return -1;
    }