题目描述
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通栈的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。
样例描述
输入:
["MyStack", "push", "push", "top", "pop", "empty"]
[[], [1], [2], [], [], []]
输出:
[null, null, null, 2, 2, false]
解释:
MyStack myStack = new MyStack();
myStack.push(1);
myStack.push(2);
myStack.top(); // 返回 2
myStack.pop(); // 返回 2
myStack.empty(); // 返回 False
思路
双队列
- 一个队列作为栈,另外一个队列作为辅助队列,用于调换元素的顺序。(因为栈里面是后进先出)。
- 入栈操作时,先将元素都加入到辅助队列,再将主队列的元素加入到辅助队列,然后交换两个队列的元素,就能够保证在主队列里的元素都是符合后到先出的。
- 容器也能用交换数的方法交换。
代码
class MyStack {
Queue<Integer> queue1, queue2;
public MyStack() {
queue1 = new LinkedList<>();
queue2 = new LinkedList<>();
}
public void push(int x) {
//先将元素加入到辅助队列
queue2.offer(x);
//将主队列的元素全部加入到辅助队里的末尾
while (!queue1.isEmpty()) {
queue2.offer(queue1.poll());
}
//交换两个队列,准备下一轮新元素插入时的迭代
Queue<Integer> t = queue1;
queue1 = queue2;
queue2 = t;
}
public int pop() {
return queue1.poll();
}
public int top() {
return queue1.peek();
}
public boolean empty() {
return queue1.isEmpty();
}
}
/**
* Your MyStack object will be instantiated and called as such:
* MyStack obj = new MyStack();
* obj.push(x);
* int param_2 = obj.pop();
* int param_3 = obj.top();
* boolean param_4 = obj.empty();
*/