题目:
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型。
思路:
队列是先进先出,栈是先进后出。
用两个栈来实现队列,可以先进A,出栈A后进入B,最终B的出栈顺序为出队列的顺序。
队列的入队push:元素入栈A
队列的出队pop:若B中有元素,则直接B.pop();若B中无元素,则将A中所有元素出栈,并入栈到B后,再进行B.pop()操作。
解答:
import java.util.Stack;
public class Solution {
Stack<Integer> stack1 = new Stack<Integer>();
Stack<Integer> stack2 = new Stack<Integer>();
//队列是先进先出,栈是先进后出。
//用两个栈来实现队列,可以先进A,出栈A后进入B,最终B的出栈顺序为出队列的顺序。
public void push(int node) {
stack1.push(node);
}
public int pop(){
if (stack2.isEmpty()){
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
}
return stack2.pop();
}
}
拓展:
如何使用两个队列实现栈呢?
思路:两个队列保持其中一个为空。
栈的push:将元素push到空的那个队列中,并将非空队列全部入队到这个队列中。(这样,另个队列就空了。)
栈的pop:将非空队列出队即可。
解答:java版本
class twoQueueToStack{
Queue<Integer> a = new LinkedList<Integer>();
Queue<Integer> b = new LinkedList<Integer>();
//保证有一个队列为空,当插入的时候,把新插入的元素放入空队列中,并出队非空队列入队该队列中
public void push(int node){
if(a.isEmpty()){
a.offer(node);
while (!b.isEmpty()){
a.offer(b.poll());
}
}else if (b.isEmpty()){
b.offer(node);
while(!a.isEmpty()){
b.offer(a.poll());
}
}else{ //两个都为空,随便插入一个
a.offer(node);
}
}
public int pop(){
if(a.isEmpty()){
return b.poll();
}else{
return a.poll();
}
}
}
网友评论