【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 。

示例 1:

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

题解:

解法1:

需要对类似[1,1,1,1,2,1,1]这样的数组恢复二分查找的二段性质,也就是需要预处理一下。[1,1,1,1,2,1,1]变为[1,1,1,1,2]。首先查找到最小值的index为l,将[0,n-1]划分为两个区间,分别为:[0,l-1]与[l,n-1]两个区间进行二分查找。

class Solution {
public:
    int search(vector<int>& nums, int target) {
        int n = nums.size();
        int l = 0, r = n - 1;
        while (l < r && nums[0] == nums[r]) {
            r--;
        }
        // cout << r << endl;
        // 查找最小值
        l = findMin(nums, l, r, nums[r]);
        // cout << l << endl;
        // nums[l] 为 最小值
        // 划分两个区间进行二分 [0,l-1] [l,r]
        if (target < nums[l]) return false;

        int x = binarySearch(nums, l, r, target);

        if (nums[x] == target)
            return true;
        int y = binarySearch(nums, 0, l-1, target);
        if (nums[y] == target)
            return true;
        return false;
    }
    int findMin(vector<int>& nums, int l, int r, int target) {
        while (l < r) {
            int mid = l + r >> 1;
            if (nums[mid] <= target) {
                r = mid;
            } else {
                l = mid + 1;
            }
        }
        return l;
    }
    int binarySearch(vector<int>& nums, int l, int r, int target) {
        while (l < r) {
            int mid = l + r >> 1;
            if (nums[mid] >= target) {
                r = mid;
            } else {
                l = mid + 1;
            }
        }
        return l;
    }
};

解法2:

与右边界比较过程中,使用mid进行比较,需要将=做单独处理,右边界--。

class Solution {
public:
    bool search(vector<int>& nums, int target) {
        int n = nums.size();
        int l = 0, r = n - 1;
        while (l < r) {
            int mid = l + r >> 1;
            if (nums[mid] == target) {
                return true;
            } else if (nums[mid] > nums[r]) { // 说明在左半边 在[0, mid-1]进行查找
                if (target >= nums[l] && target < nums[mid]) {
                    r = mid;
                } else {
                    l = mid + 1;
                }
            } else if (nums[mid] < nums[r]) {
               if (target > nums[mid] && target <= nums[r]) {
                   l = mid + 1;
               } else {
                   r = mid;
               }
            } else { // 处理重复元素
                r--;
            }
        }

        if (nums[l] == target) return true;
        return false;
    }
};

本节完~

【LeetCode每日一题】81. 搜索旋转排序数组 II_streaming