Leetcode 33搜索旋转排序数组 Search in Rotated Sorted Array

题目：整数数组 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 。

你必须设计一个时间复杂度为 O(log n) 的算法解决此问题。
link

题解

• search题目，复杂度O(logn)——想到二分查找
• 原先是ascending order，所以旋转后也会是2个升序排列的，且前面的升序大
• 二分先考虑[mid]是在前一个升序中，还是后一个中
  • 在前一个中 …
  • 在后一个中 …

具体见下图

note: 注意边界！记得使用模板的区间是[l, r)，左闭右开。

代码

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int n = nums.size();
        if (n == 1)
            return (target == nums[0]) ? 0 : -1;    // 只有1个数，是target 返回index=0

        int l = 0, r = n;
        while (l < r) {
            int mid = (l + r) >> 1;
            if (nums[mid] == target)
                return mid;
            if (nums[mid] >= nums[0]) {
                if (nums[mid] > target && nums[0] <= target)
                    r = mid;
                else
                    l = mid + 1;
            }
            else {
                if (nums[mid] < target && target <= nums[n - 1])
                    l = mid + 1;
                else
                    r = mid;
            }
        }
        return -1;
    }
};