题目
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数appendTail和deleteTail,分别完成在队列尾部插入节点和在队列头部移除节点的功能。
分析
队列的性质是先进先出
栈的性质是先进后出
我们用两个栈stack1和stack2实现一个队列。
插入元素:我们把元素插入stack1中。首先插入a,如果还有元素,再接着插入,比如再插入b,c
删除元素:a在b之前,b在c之前插入的,所有删除元素时,应该是先删除a,然后删除b,在删除c。对于stack1,出栈的顺序和入栈的顺序是相反的。即从stack1中弹出时的顺序是c,b,a。这时候我们就要用到另一个栈stack2。我们把从stack1中弹出的c,b,a,依次入到stack2中。再从stack2中弹出元素。这时弹出元素的顺序是a,b,c,符合我们对队列的先进先出的要求。
总结一下:当stack2不为空时,在stack2中的栈顶元素是最先进入队列的元素,可以弹出。当stack2为空时,把stack1中的元素逐个弹出并压入stack2,然后可以从stack2中弹出元素。
代码
public class TwoStackSimulateQueue {
Stack<Integer> stack1 = new Stack<>();
Stack<Integer> stack2 = new Stack<>();
public void appendTail(int a) {
stack1.push(a);
}
public int deleteTail(){
if(stack2.isEmpty()){
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
}
if(stack2.size() == 0) {
return -1;
}
return stack2.pop();
}
}