力扣 33. 搜索旋转排序数组

🔗 https://leetcode.cn/problems/search-in-rotated-sorted-array

题目

• 给一个数组，满足 nums[0]… nums[k] 和 nums[k+1]…. nums[n] 都是升序，且 nums[n] < nums[0]
• 给一个 target 数字，返回在 nums 中的下标，不存在返回 -1

思路

• 比较特殊的二分，需要考虑到 nums[mid] 所处的序列
• 若 nums[l] … nums[mid] 是升序序列
  • 若 target 在 nums[l] … nums[mid] 这个序列之中，r = mid - 1
  • 若 target 不在 nums[l] … nums[mid] 这个序列之中，剩余的部分是要继续二分的，l = mid+1
• 若 nums[l] … numd[mid] 中间有两个升序序列
  • 若 target 在 nums[mid]… nums[r] 这个序列之中，l = mid +1
  • 若 target 不在 nums[mid]… nums[r] 这个序列之中，剩余的部分是要继续二分的，r = mid - 1

代码

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

        while (l <= r) {
            int mid = (l + r) /2;
            if (nums[mid] == target) return mid;

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

        }
        return -1;
        
    }
};