1. 问题描述:
给你一个整数数组 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 * 10 ^ 5
-10 ^ 9 <= nums[i] <= 10 ^ 9
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/132-pattern
2. 思路分析:
这道题目的思路是比较难想的,在y总的视频中使用的是单调栈的做法。首先我们考虑怎么样枚举才可以将答案枚举出来,也即首先考虑做法,其次考虑借助于数据结构进行优化,这里可以从后往前进行枚举ai,我们在枚举的时候是需要找到组合(aj,ak)使得ai < ak < aj,ai与ak才有直接的关系,所以我们在从后往前枚举ai的时候找到满足要求的ak即可,这里满足要求的ak的意思是存在一个aj,使得aj大于ak,基于这个特点想到使用单调栈来维护这个ak使得在枚举ai的过程中快速找到满足要求的ak,使用一个int类型的变量right来维护当前ai右边满足要求的最大的ak,在单调栈维护的过程中弹出栈顶当前比nums[i]小的元素,弹出的元素就是满足要求的ak(nums[i]相当于是aj),在弹出栈顶元素的时候维护right的最大值,表示满足当前条件最大的ak,这一点很巧妙,等到下一次枚举ai的时候如果发现当前right大于ai,说明存在满足要求的ak使得ai < ak < aj,返回True即可,循环结束之后说明没有找到答案返回False。结合例子[3, 1, 4, 2]和debug调试工具会比较好理解。
3. 代码如下:
from typing import List
import sys
class Solution:
def find132pattern(self, nums: List[int]) -> bool:
stack = list()
n = len(nums)
right = -sys.maxsize
for i in range(n - 1, -1, -1):
# 枚举的是ai当right>ai之后说明找到了ai < ak < aj的组合
if right > nums[i]: return True
while stack and stack[-1] < nums[i]:
# right维护的就是满足条件的ak, 为当前ai右边最大的满足要求的ak
right = max(right, stack[-1])
stack.pop()
stack.append(nums[i])
return False