【LeetCode-81】搜索旋转排序数组 II

问题

已知存在一个按非降序排列的整数数组 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,4,4,5,6,6,7] 在下标 5 处经旋转后可能变为 [4,5,6,6,7,0,1,2,4,4] 。

给你 旋转后 的数组 nums 和一个整数 target ，请你编写一个函数来判断给定的目标值是否存在于数组中。如果 nums 中存在这个目标值 target ，则返回 true ，否则返回 false 。

示例

输入： nums = [2,5,6,0,0,1,2], target = 0
输出： true

解答

class Solution {
public:
    bool search(vector<int>& nums, int target) {
        int n = nums.size();
        int left = 0, right = n - 1;
        while (left < right && nums[0] == nums[right]) right--; // 恢复数组的“二段性”
        n = right + 1; // 具备二段性的数组长度
        while (left < right) {
            int mid = (left + right) / 2;
            if (nums[mid] > nums[right]) left = mid + 1;
            else right = mid;
        }
        right = nums[n - 1] < target ? left - 1 : n - 1; // 判断在左还是在右
        left = nums[n - 1] < target ? 0 : left;
        while (left < right) {
            int mid = (left + right) / 2;
            if (nums[mid] < target) left = mid + 1;
            else right = mid;
        }
        return nums[left] == target;
    }
};

重点思路

与【LeetCode-33】搜索旋转排序数组的代码就两行差别。两题的区别为，本题含有重复数字。“二段性”的定义为，只要一段数组满足一个性质，那么剩下的一段数组一定不满足这个性质。当含有重复数字是，例如[2,5,6,0,0,1,2]这样的例子，target = 2可能同时存在于左右两个数组中。要使用二分查找，必须恢复数组的二段性。方法也很简单，就是将数组末尾与数组开头相同的数据删除就行。