1. 题目
请你仅使用两个队列实现一个后入先出(LIFO)的栈,并支持普通队列的全部四种操作(push、top、pop 和 empty)。
实现 MyStack 类:
void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的,返回 true ;否则,返回 false 。
注意:
你只能使用队列的基本操作 —— 也就是 push to back、peek/pop from front、size 和 is empty 这些操作。
你所使用的语言也许不支持队列。 你可以使用 list (列表)或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
示例:
输入:
[“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
提示:
1 <= x <= 9
最多调用100 次 push、pop、top 和 empty
每次调用 pop 和 top 都保证栈不为空
进阶:你能否实现每种操作的均摊时间复杂度为 O(1) 的栈?换句话说,执行 n 个操作的总时间复杂度 O(n) ,尽管其中某个操作可能需要比其他操作更长的
时间。你可以使用两个以上的队列。
Related Topics 栈 设计
👍 338 👎 0
2. 题解
2.1 解法1: 一个队列
每次push前先记录队列长度 n, 然后将元素加入队列, 最后将 n 个元素出队再加入队列
class MyStack {
Queue<Integer> queue;
/**
* Initialize your data structure here.
*/
public MyStack() {
queue = new LinkedList<>();
}
/**
* Push element x onto stack.
*/
public void push(int x) {
int size = queue.size();
queue.offer(x);
for (int i = 0; i < size; i++) {
int temp = queue.poll();
queue.offer(temp);
}
}
/**
* Removes the element on top of the stack and returns that element.
*/
public int pop() {
return queue.poll();
}
/**
* Get the top element.
*/
public int top() {
return queue.peek();
}
/**
* Returns whether the stack is empty.
*/
public boolean empty() {
return queue.isEmpty();
}
}
2.2 解法2: 两个队列
使用一个辅助队列, 每次 push 直接将当前元素加入辅助队列, 然后将主队列所有元素出队加入辅助队列, 最后交换主队列与辅助队列
class MyStack {
Queue<Integer> mainQue;
Queue<Integer> secondQue;
/**
* Initialize your data structure here.
*/
public MyStack() {
mainQue = new LinkedList<>();
secondQue = new LinkedList<>();
}
/**
* Push element x onto stack.
*/
public void push(int x) {
secondQue.offer(x);
while (!mainQue.isEmpty()) {
secondQue.offer(mainQue.poll());
}
Queue<Integer> temp = mainQue;
mainQue = secondQue;
secondQue = temp;
}
/**
* Removes the element on top of the stack and returns that element.
*/
public int pop() {
return mainQue.poll();
}
/**
* Get the top element.
*/
public int top() {
return mainQue.peek();
}
/**
* Returns whether the stack is empty.
*/
public boolean empty() {
return mainQue.isEmpty();
}
}
参考:
官方题解