文章目录
前言
记录自己刷题思路,时常复习。
`
一、栈与队列
剑指offer 09.用两个栈实现队列
剑指offer 30. 包含min函数的栈
二、解题思路
1.用两个栈实现队列
解题思路:栈的特性是后进先出,而队列的特性是先进先出,用两个栈实现队列,则考虑一个栈用来实现“入队操作。另一个栈实现出队操作。
首先定义两个栈,入队操作栈 stack_in, 出队操作栈 stack_out
每次入队时,stack_in 正常入栈, 出队时,要判断stack_out是否为空,若为空则将stack_in的元素全部出栈加入到stack_out栈中,即可调转先后顺序,之后在stack_out栈pop操作即可。
python3代码如下:
from collections import deque
class CQueue:
def __init__(self):
self.stack1 = deque()
self.stack2 = deque()
def appendTail(self, value: int) -> None:
self.stack1.append(value)
def deleteHead(self) -> int:
if self.stack2:
return self.stack2.pop()
else:
if self.stack1:
while self.stack1:
self.stack2.append(self.stack1.pop())
return self.stack2.pop()
else:
return -1
# Your CQueue object will be instantiated and called as such:
# obj = CQueue()
# obj.appendTail(value)
# param_2 = obj.deleteHead()
2.包含min函数的栈
**解题思路:**本题与上一题思路大致相同,都需要用到两个栈,其中一个栈作为辅助栈。**需要注意的是:**题目要求调用min、push以及pop的时间复杂度都是O(1)。
首先定义两个栈stack、stack_min;一个作为主栈一个作为记录当前状态下最小值的辅助栈。
当有元素要入栈时,首先需要判断:stack_min栈是否为为空,若为空则当前元素即为最小值,若不为空,则需要判断当前元素是否小于min栈顶元素,若小于,则入stack_min栈push该元素。
当有元素pop出栈时,则需要判断当前元素是否与min栈顶元素相等,如果相等stack_min栈也执行pop操作。
python3代码如下:
from collections import deque
class MinStack:
def __init__(self):
self.stack = deque()
self.stack_m = deque()
def push(self, x: int) -> None:
self.stack.append(x)
if self.stack_m:
if x <= self.stack_m[-1]:
self.stack_m.append(x)
else:
self.stack_m.append(x)
def pop(self) -> None:
if self.stack.pop() == self.stack_m[-1]:
self.stack_m.pop()
def top(self) -> int:
return self.stack[-1]
def min(self) -> int:
return self.stack_m[-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()
总结
一个栈和队列的简单操作,记录一下,时常复习