LeetCode 81. Search in Rotated Sorted Array II

最新推荐文章于 2022-05-27 18:33:05 发布

原创最新推荐文章于 2022-05-27 18:33:05 发布 · 297 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#有序翻转数组查找 #search #sorted #leetcode

LeetCode 同时被 2 个专栏收录

98 篇文章

订阅专栏

LeetCode

98 篇文章

订阅专栏

本文讨论了含有重复元素的旋转有序数组搜索问题，并提供了一种改进的二分查找算法实现。文章详细解释了如何通过判断子数组是否有序来决定搜索方向，并针对重复元素采取特殊处理策略。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述

Follow up for "Search in Rotated Sorted Array":
What if duplicates are allowed?

Would this affect the run-time complexity? How and why?

Suppose an array sorted in ascending order is rotated at some pivot unknown to you beforehand.

(i.e., 0 1 2 4 5 6 7 might become 4 5 6 7 0 1 2).

Write a function to determine if a given target is in the array.

The array may contain duplicates.

分析
允许重复元素，则上一题中如果 A[m]>=A[l], 那么 [l,m] 为递增序列的假设就不能成立了，比如 [1,3,1,1,1]。
如果 A[m]>=A[l] 不能确定递增，那就把它拆分成两个条件:
• 若 A[m]>A[l]，则区间 [l,m] 一定递增
• 若 A[m]==A[l] 确定不了，那就 l++，往下看一步即可。

代码

class Solution {
public:
    bool search(vector<int>& nums, int target) {
        int first = 0;
        int last = nums.size();

        while (first != last) {
            int mid = (first + last) / 2;

            if (nums[mid] == target)
                return true;

            if (nums[first] < nums[mid]) {
                if (nums[first] <= target && target < nums[mid])
                    last = mid;
                else
                    first = mid + 1;
            }
            else if (nums[first] > nums[mid]) {
                if (nums[mid] < target && target <= nums[last - 1])
                    first = mid + 1;
                else
                    last = mid;
            }
            else 
                ++first;            // skip duplicate one
        }
        return false;
    }
};