[S]O-30/31包含min函数的栈/栈的压入弹出序列
1.1O-30包含min函数的栈
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
1.2分析与代码
思路:
普通栈的 push() 和 pop() 函数的复杂度为 O(1);而获取栈最小值 min() 函数需要遍历整个栈,复杂度为 O(N) 。
本题难点: 将 min() 函数复杂度降为 O(1) ,可通过建立辅助栈实现;
数据栈 A : 栈 A 用于存储所有元素,保证入栈 push() 函数、出栈 pop() 函数、获取栈顶 top() 函数的正常逻辑。
辅助栈 B : 栈 B 中存储栈 A中所有 非严格降序 的元素,则栈 A中的最小元素始终对应栈 B的栈顶元素,即 min() 函数只需返回栈 B的栈顶元素即可。
因此,只需设法维护好 栈 B的元素,使其保持非严格降序,即可实现 min() 函数的 O(1)复杂度。
- push(x) 函数: 重点为保持栈 B 的元素是 非严格降序 的。
将 x 压入栈 A (即 A.add(x) );
若 ① 栈 B为空 或 ② x小于等于 栈 B 的栈顶元素,则将 x 压入栈 B(即 B.add(x) )。 - pop() 函数: 重点为保持栈 A, B的 元素一致性 。
执行栈 A 出栈(即 A.pop() ),将出栈元素记为 y ;
若 y 等于栈 B 的栈顶元素,则执行栈 B 出栈(即 B.pop() )。 - top() 函数: 直接返回栈 A的栈顶元素
- min() 函数: 直接返回栈 B 的栈顶元素
参考链接:link
class MinStack:
def __init__(self):
"""
initialize your data structure here.
"""
self.A,self.B=[],[]
def push(self, x: int) -> None:
self.A.append(x)
if not self.B or self.B[-1]>=x:
self.B.append(x)
def pop(self) -> None:
if self.A.pop()==self.B[-1]:
self.B.pop()
def top(self) -> int:
return self.A[-1]
def min(self) -> int:
return self.B[-1]
# Your MinStack object will be instantiated and called as such:
# obj = MinStack()
# obj.push(x)
# obj.pop()
# param_3 = obj.top()
# param_4 = obj.min()
2.1O-31包含min函数的栈/栈的压入弹出序列
输入两个整数序列,第一个序列表示栈的压入顺序,请判断第二个序列是否为该栈的弹出顺序。假设压入栈的所有数字均不相等。例如,序列 {1,2,3,4,5} 是某栈的压栈序列,序列 {4,5,3,2,1} 是该压栈序列对应的一个弹出序列,但 {4,3,5,1,2} 就不可能是该压栈序列的弹出序列。
输入:pushed = [1,2,3,4,5], popped = [4,5,3,2,1]
输出:true
解释:我们可以按以下顺序执行:
push(1), push(2), push(3), push(4), pop() -> 4,
push(5), pop() -> 5, pop() -> 3, pop() -> 2, pop() -> 1
输入:pushed = [1,2,3,4,5], popped = [4,3,5,1,2]
输出:false
解释:1 不能在 2 之前弹出。
2.2分析与代码
栈的数据操作具有 先入后出 的特性,因此某些弹出序列是无法实现的。
考虑借用一个辅助栈 stackstack ,模拟 压入 / 弹出操作的排列。根据是否模拟成功,即可得到结果。
入栈操作: 按照压栈序列的顺序执行。
出栈操作: 每次入栈后,循环判断 “栈顶元素 弹出序列的当前元素” 是否成立,将符合弹出序列顺序的栈顶元素全部弹出。
由于题目规定 栈的所有数字均不相等 ,因此在循环入栈中,每个元素出栈的位置的可能性是唯一的(若有重复数字,则具有多个可出栈的位置)。因而,在遇到 “栈顶元素 == 弹出序列的当前元素” 就应立即执行出栈。
流程:
- 初始化
辅助栈 stack ,弹出序列的索引 ii ; - 遍历压栈序列
各元素记为 num ;
元素 num入栈; - 循环出栈
若 stack的栈顶元素 == 弹出序列元素 popped[i] ,则执行出栈与 i++; - 返回值
若 stack为空,则此弹出序列合法。
class Solution:
def validateStackSequences(self, pushed: List[int], popped: List[int]) -> bool:
stack,i=[],0
for num in pushed:
stack.append(num)
while stack and stack[-1]==popped[i]:
stack.pop()
i+=1
return not stack
参考链接ppt动画:link
----------2021/01/11-栈相关一般是辅助栈---------