LeetCode 33. 搜索旋转排序数组(二分)

问题描述

升序排列的整数数组 nums 在预先未知的某个点上进行了旋转（例如， [0,1,2,4,5,6,7] 经旋转后可能变为 [4,5,6,7,0,1,2] ）。

请你在数组中搜索 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

提示：

1 <= nums.length <= 5000
-10^4 <= nums[i] <= 10^4
nums 中的每个值都 独一无二
nums 肯定会在某个点上旋转
-10^4 <= target <= 10^4

思路

由于数组分成了两个有序区间，第一步二分找出使数组分成两个有序区间的分界点，第二步判断target在哪个区间，第三步二分找出目标值。(二分时要注意边界).

AC代码

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int l = 0,r = nums.size() - 1;

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

        if (target >= nums[0]) l = 0;
        else r = nums.size() - 1;

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

        if (target == nums[l]) return l;
        else return -1;
    }
};