给你一个整数数组 nums ,数组中共有 n 个整数。132 模式的子序列 由三个整数 nums[i]、nums[j] 和 nums[k] 组成,并同时满足:i < j < k 和 nums[i] < nums[k] < nums[j] 。
如果 nums 中存在 132 模式的子序列 ,返回 true ;否则,返回 false 。
示例 1:
输入:nums = [1,2,3,4]
输出:false
解释:序列中不存在 132 模式的子序列。
示例 2:
输入:nums = [3,1,4,2]
输出:true
解释:序列中有 1 个 132 模式的子序列: [1, 4, 2] 。
示例 3:
输入:nums = [-1,3,2,0]
输出:true
解释:序列中有 3 个 132 模式的的子序列:[-1, 3, 2]、[-1, 3, 0] 和 [-1, 2, 0] 。
提示:
n == nums.length
1 <= n <= 2 * 105
-109 <= nums[i] <= 109
这道题乍一看挺简单,三层遍历时间复杂度n3,妥妥的超时,所以这里采用单调栈存储可能的nums[k]的方法。
我们遍历i,j,k中的j,即在nums[j]前找一nums[i]使其小于nums[j],在nums[j]后找一数nums[k]使其大于nums[j](其实用i,k同可以做,方便起见这里以j为例,大家可以自己试试其他两种方法)。
基本思路如下:
从后往前遍历nums,并将值存储在单调栈中,此单调栈存储可能的nums[k]的取值。若栈顶元素(min)小于即将遍历的nums[i],则说明这个nums[i]有作为nums[j]的潜质,依次弹出小于该nums[i]中的元素,取其中的最大值即为满足条件最大的nums[k],若在之后的遍历中存在nums[i]>nums[j],则进行nums[j]和nums[k]的更新,若存在nums[i]<nums[k],则判断为满足条件,返回true。
直接描述可能有点抽象,下面以nums = [1,3,2,4,5,6]为例,首先从后向前遍历nums,将值存储到单调栈stack中,因为开始时遍历的nums[i]均小于最后的nums[k](即nums[-1],此时nums[k]还未更新),直至stack中元素为[6,5,4,2],此时nums[i]为3大于栈顶元素2,故从stack中弹出2并将值付给nums[k],之后栈顶元素变为4,所以满足条件的最大nums[k]为2,继续遍历nums,此时nums[i]为1小于nums[k],故找到满足条件的组合1,3,2
代码如下,yysy刚开始时没咋想用单调栈append pop写,结果越写越迷把自己绕进去了…可见数据结构算法的重要性
class Solution:
def find132pattern(self, nums: List[int]) -> bool:
stack = []
k = float("-inf")
for i in range(len(nums) - 1,-1,-1):
if nums[i] < k:
return True
while stack and stack[-1] < nums[i]:
k = max(k,stack.pop())
stack.append(nums[i])
return False