leetcode-----33. 搜索旋转排序数组

Posted 2020-12-12 景云

• 链接：https://leetcode-cn.com/problems/search-in-rotated-sorted-array/

思路

找出两段；通过找出满足条件：第一段的数字都大于第一个数。
第一次二分找出分段点，第二次二分求解。

代码

class Solution {
public:
    int search(vector<int>& nums, int target) {
        if (nums.empty()) return -1;
        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 {
            l = r + 1, r = nums.size() - 1;
        }

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

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