24

LeetCode 81,在不满足二分的数组内使用二分法 II

 4 years ago
source link: http://www.cnblogs.com/techflow/p/13235840.html
Go to the source link to view the article. You can view the picture content, updated content and better typesetting reading experience. If the link is broken, please click the button below to view the snapshot at that time.
neoserver,ios ssh client

今天是 LeetCode专题 第50篇文章,我们来聊聊LeetCode中的81题Search in Rotated Sorted ArrayII。

它的官方难度是 Medium ,点赞1251,反对470,通过率32.8%。从通过率上来看,这题属于Medium难度当中偏难一些的题目,也的确如此,稍稍有些考验思维。

题意

假设我们有一个含有重复元素的有序数组,我们随意选择一个位置将它分成两半,然后将这两个部分调换顺序拼接成一个新的数组。现在给定一个target,要求返回一个bool结果, 表明target是否在数组当中

样例

Input: nums = [2,5,6,0,0,1,2], target = 0
Output: true
Input: nums = [2,5,6,0,0,1,2], target = 3
Output: false

如果是你按照顺序刷LeetCode或者是本专题的话,你会发现我们在之前做过一道非常相似的题目。它就是 LeetCode的33题 ,Search in Rotated Sorted ArrayI。不过不同的是,在33题的题意当中,明确表明了数组当中的元素是不包含重复元素的,除此之外,这两题的题意完全一样。

LeetCode 33,在不满足二分的数组内使用二分的方法

这么一点小小的差别会带来解法的变化吗?

题解

答案当然是肯定的,不然出题人可以退休了。

问题是,问题出在哪里呢?

我们先不着急,先来回忆一下33题中的做法。我们当时使用了一个最简单的笨办法,就是 先通过二分法找到数组截断的位置 。然后再通过截断的位置还原出原数组的情况,根据我们target的大小,找到它可能存在的位置。

但是在当前这个问题当中,这个思路走不通了。走不通的原因也很简单,就是因为 重复元素 的存在。

举个例子:[1, 3, 1, 1, 1, 1, 1, 1]

当我们进行二分查找的时候,发现mid是1和left的1相等,我们根本 无法判断截断点究竟在mid的左侧还是右侧 ,二分查找也就无从谈起了。

我们当然可以退一步采用遍历的方法去寻找切分点,但是既然如此,我们为什么不直接去寻找答案呢?反正都已经是O(n)的复杂度了。所以这是行不通的,我们想要使得复杂度维持在 就必须要寻找其他的路数。

思路和解法很多时候不是凭空来的,需要我们对问题进行深入的分析。在这个问题当中,我们的问题是明确并且简单的。就是一个调换了部分顺序的有序数组,只是我们不确定的是调换的部分究竟有多长。由于我们最终希望 通过二分法来寻找答案 ,所以我们可以根据调换的元素是否过半想出两种情况来。

我把这两种情况用图展示出来:

7bAVviN.jpg!web

也就是说我们的 分割点可能在数组的前半段也可能在后半段 ,对于这两种情况我们的处理方法是不同的。

我们先看第一种情况,数组的前半段是有序的,后半段存在截断。如果target的范围在前半段当中,我们可以抛弃掉后半段,直接在前半段中进行二分。否则,我们需要舍弃前半段,在后半段当中重复这个过程。我们可以把后半段看成是一个全新的问题,也一样可以分成两种情况,类似于递归一样的往下执行即可。

再来看第二种情况,第二种情况的后半段和第一种情况的前半段是一样的,都是有序的元素,我们直接二分即可。它的前半段和第一种情况的后半段是一样的,我们没法判断,需要继续二分。

也就是说,我们 只能在有序的数组进行二分 ,如果当前数组存在分段,不是整体有序的,那我们就对它进行拆分。拆分之后总能找到有序的部分,如果还找不到就继续拆分。因为分段点只有一个,所以不论当前的数组什么样,拆分一次之后, 必然至少可以找到一段是有序的

想明白这点之后就简单了,看起来很像是递归,但实际上它的本质仍然是二分。代码并不难写,但是还有一个问题没解决,就是当nums[m] = nums[l]的时候,我们如何判断是哪一种情况呢?

答案是没法判断,两种情况都有可能,对于这种情况也没有很好的办法,我想出来的办法是可以将l向右移动一位,相当于 抛弃了一个最左侧的数 。我们把这些思路总结总结,代码也就出来了:

class Solution:
    def search(self, nums: List[int], target: int) -> bool:
        l, r = 0, len(nums)-1
        while l <= r:
            m = (l + r) >> 1
            if nums[m] == target:
                return True
            
            if nums[l] == nums[m]:
                l += 1
                continue
                
            if nums[l] < nums[m]:
                if nums[l] <= target < nums[m]:
                    r = m - 1
                else:
                    l = m + 1
            else:
                if nums[m] < target <= nums[r]:
                    l = m + 1
                else:
                    r = m - 1
        return False

总结

到这里,我们关于这道题的题解就结束了。在问题的最后,出题人给我们留了一个问题,和33题比起来,这题的解法的 时间复杂度会有变化吗

表面上看我们一样用到了二分,所以同样是log级的复杂度,问题的复杂度并没有变化。但实际上并不是这样的,我们来看一种最坏的情况, 假设数组当中所有的值全部相等 。这个时候二分就不起效果了,最终会退化成O(n)的线性枚举,这样又变成了O(n)的复杂度。当然,在大部分情况下,这并不会发生。所以是算法的最坏复杂度退化成了O(n),平均复杂度依然是O(logN)。

如果喜欢本文,可以的话,请 点个关注 ,给我一点鼓励,也方便获取更多文章。

本文使用 mdnice 排版


About Joyk


Aggregate valuable and interesting links.
Joyk means Joy of geeK