本文始发于个人公众号:TechFlow,原创不易,求个关注
今天是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题的题意当中,明确表明了数组当中的元素是不包含重复元素的,除此之外,这两题的题意完全一样。
这么一点小小的差别会带来解法的变化吗?
题解
答案当然是肯定的,不然出题人可以退休了。
问题是,问题出在哪里呢?
我们先不着急,先来回忆一下33题中的做法。我们当时使用了一个最简单的笨办法,就是先通过二分法找到数组截断的位置。然后再通过截断的位置还原出原数组的情况,根据我们target的大小,找到它可能存在的位置。
但是在当前这个问题当中,这个思路走不通了。走不通的原因也很简单,就是因为重复元素的存在。
举个例子:[1, 3, 1, 1, 1, 1, 1, 1]
当我们进行二分查找的时候,发现mid是1和left的1相等,我们根本无法判断截断点究竟在mid的左侧还是右侧,二分查找也就无从谈起了。
我们当然可以退一步采用遍历的方法去寻找切分点,但是既然如此,我们为什么不直接去寻找答案呢?反正都已经是O(n)的复杂度了。所以这是行不通的,我们想要使得复杂度维持在就必须要寻找其他的路数。
思路和解法很多时候不是凭空来的,需要我们对问题进行深入的分析。在这个问题当中,我们的问题是明确并且简单的。就是一个调换了部分顺序的有序数组,只是我们不确定的是调换的部分究竟有多长。由于我们最终希望通过二分法来寻找答案,所以我们可以根据调换的元素是否过半想出两种情况来。
我把这两种情况用图展示出来:
也就是说我们的分割点可能在数组的前半段也可能在后半段,对于这两种情况我们的处理方法是不同的。
我们先看第一种情况,数组的前半段是有序的,后半段存在截断。如果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 排版