使用栈实现队列的下列操做:java
Implement the following operations of a queue using stacks.python
示例:编程
MyQueue queue = new MyQueue();
queue.push(1);
queue.push(2);
queue.peek(); // 返回 1
queue.pop(); // 返回 1
queue.empty(); // 返回 false
复制代码
说明:数组
push to top
, peek/pop from top
, size
, 和 is empty
操做是合法的。Notes:bash
push to top
, peek/pop from top
, size
, and is empty
operations are valid.队列先进后出,栈后进先出。用栈实现队列,能够用两个栈完成题解。入队列时用 stack1 存入节点,出队列时 stack1 内节点顺序出栈压入 stack2 中。数据结构
例如 1, 2, 3 元素顺序入队列
即存入栈stack1:[1, 2, 3]
出队列时顺序应为:1->2->3
可是栈先进先出,出栈顺序为:3->2->1
与出队列顺序不相符
借助另外一个栈stack2
stack1内的元素顺序出栈并压入stack2
stack1:[1, 2, 3] ---> stack2:[3, 2, 1]
此时stack2出栈顺序:1->2->3
与出队列顺序相符
复制代码
**注意:**在出队列时无需着急将 stack1 中的节点顺序压入 stack2。由于要实现的队列是先进后出,能够将 stack2 中的节点所有弹出以后 再将 stack1 内节点顺序压入stack2,这样能够将出栈的时间复杂度摊还到 O(1)。app
class MyQueue {
private Stack<Integer> stack1;
private Stack<Integer> stack2;
public MyQueue() {
stack1 = new Stack<>();
stack2 = new Stack<>();
}
public void push(int x) {
stack1.push(x);
}
public int pop() {
if (stack2.isEmpty()) {
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
public int peek() {
//stack1节点顺序弹出并压入stack2
if (stack2.isEmpty()) {//条件是: stack2为空,而不是stack1非空, 这样摊还复杂度O(1)
while (!stack1.isEmpty()) {
stack2.push(stack1.pop());//stack1弹出节点并压入stack2
}
}
return stack2.peek();
}
public boolean empty() {
return stack1.isEmpty() && stack2.isEmpty();
}
}
复制代码
Python语言没有栈和队列数据结构,只能用数组 List 或双端队列 deque 实现。编程语言
这类编程语言就压根不须要 用队列实现栈或用栈实现队列这种问题,由于栈和队列自己就必须借助List、deque实现。spa
因此这道题在这种语言中这就很是简单了,能够说是做弊。code
class MyQueue:
def __init__(self):
self.queue = []
def push(self, x: int) -> None:
self.queue.append(x)
def pop(self) -> int:
#弹出第一个元素
return self.queue.pop(0)
def peek(self) -> int:
#返回第一个元素
return self.queue[0]
def empty(self) -> bool:
return not self.queue
复制代码
欢迎关注微.信公.众号:爱写Bug