题目:
请你仅使用两个队列实现一个后入先出(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(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
解题思路:
- 入栈的时候,入到不为空的队列,刚开始两个队列都为空的话,指定入到一个队列;
- 出栈的时候,找到不为空的队列,出size-1个元素到另一个队列中去,剩下的这个元素就是出栈的元素;
关键是画图分析:
具体代码实现如下 :
class MyStack {
public Queue<Integer> qu1;
public Queue<Integer> qu2;
public MyStack() {
qu1=new LinkedList<>();
qu2=new LinkedList<>();
}
public void push(int x) { //入栈模拟实现(qu1+qu2)
if (!qu1.isEmpty()){
qu1.offer(x);
}else if (!qu2.isEmpty()){
qu2.offer(x);
}else{
qu1.offer(x);
}
}
public int pop() { //出栈模拟实现(qu1+qu2)
if (empty()) return -1;
if (!qu1.isEmpty()){
int size=qu1.size();
for (int i = 0; i < size-1; i++) {
int val=qu1.poll();
qu2.offer(val);
}
return qu1.poll();
}
if (!qu2.isEmpty()){
int size=qu2.size();
for (int i = 0; i < size-1; i++) {
int val=qu2.poll();
qu1.offer(val);
}
return qu2.poll();
}
return -1;
}
public int top() { //返回栈顶元素:与上面出栈类似,只不过遍历时候size为全部遍历,最后存到val里面的元素就是栈顶元素!
if (empty()) return -1;
if (!qu1.isEmpty()){
int size=qu1.size();
int val=-1;
for (int i = 0; i < size; i++) {
val=qu1.poll();
qu2.offer(val);
}
return val;
}
if (!qu2.isEmpty()){
int size=qu2.size();
int val=-1;
for (int i = 0; i < size; i++) {
val=qu2.poll();
qu1.offer(val);
}
return val;
}
return -1;
}
public boolean empty() { //两个队列都为空的时刻Mystack 为空
return qu1.isEmpty() && qu2.isEmpty();
}
}