232. 用栈实现队列(easy)
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):
实现 MyQueue 类:
void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空,返回 true ;否则,返回 false
思路
使用栈来模拟队列的难点在于如何用 先进后出 来模拟 先进先出 ,对于puch可以直接用append来实现,对于empty更是简单,难点在于pop和peek的实现。
- 构建两个空栈stack_in和stack_out,前者负责输入工作,后者负责输出工作。
- puch时,直接将元素append到stack_in中。
- pop时,期望得到stack_in[0],如果直接用stack_in.pop()会返回stack_in[-1]。只需要先将stack_in依次pop到stack_out中,此时stack_in[0]就跑到了stack_out[-1]的位置,现在再stack_out.pop()即可得到答案。
- peek时,只需要pop一下,再把pop出去的值append到stack_out中
class MyQueue:
def __init__(self):
self.stack_in = []
self.stack_out = []
def push(self, x: int) -> None:
self.stack_in.append(x)
def pop(self) -> int:
if self.empty():
return None
if self.stack_out: # 如果输出栈中有元素,就直接pop
return self.stack_out.pop()
else: # 如果没有,就把输入栈中所有元素全部扔到输出栈里,然后pop
for i in range(len(self.stack_in)):
self.stack_out.append(self.stack_in.pop()) # 此时输入栈的元素全部跑到输出栈
return self.stack_out.pop()
def peek(self) -> int:
ret = self.pop()
self.stack_out.append(ret)
return ret
def empty(self) -> bool:
return not(self.stack_in or self.stack_out)
225. 用队列实现栈(easy)
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。
思路1 直接用数组得了
class MyStack:
def __init__(self):
self.list = []
def push(self, x: int) -> None:
self.list.append(x)
def pop(self) -> int:
return self.list.pop()
def top(self) -> int:
ans = self.list.pop()
self.list.append(ans)
return ans
def empty(self) -> bool:
return not self.list
思路2 老老实实用队列
class MyStack:
def __init__(self):
self.queue1 = collections.deque()
self.queue2 = collections.deque()
def push(self, x: int) -> None:
self.queue2.append(x)
while self.queue1:
self.queue2.append(self.queue1.popleft())
self.queue1, self.queue2 = self.queue2, self.queue1
def pop(self) -> int:
return self.queue1.popleft()
def top(self) -> int:
return self.queue1[0]
def empty(self) -> bool:
return not self.queue1