问题描述:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead
,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
示例 1:
输入:
["CQueue","appendTail","deleteHead","deleteHead"]
[[],[3],[],[]]
输出:[null,null,3,-1]
示例 2:
输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]
网站链接:https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
解题思路:
用栈实现队列就是想办法让新进来的元素,处于栈的底部;所以当添加元素时,我们先将原有的元素放到另一个辅助栈中,则可以把新元素放入栈底;然后再把辅助栈中的元素放回栈中。这样就可以使得新进来的元素处于栈底,最先进来的元素处于栈顶;当要删除元素时,直接从站顶中移除元素即可。
代码实现:
import java.util.Stack;
public class t08用两个栈实现队列 {
Stack<Integer> a;
Stack<Integer> b; //用于辅助
int size = 0; //记录队列的元素个数
public t08用两个栈实现队列() {
a = new Stack<Integer>();
b = new Stack<Integer>();
}
//进入队列
public void appendTail(int value) {
//将a中的元素暂时先放到b中, 使新添加的元素放到栈的底部
while(!a.isEmpty()){
b.push(a.pop());
}
a.push(value);
//将b中的元素还回去
while(!b.isEmpty()){
a.push(b.pop());
}
size++;
}
//移出队列
public int deleteHead() {
if(size<=0){
return -1;
}
int res = a.pop();
size--; //使用pop()后,再进行自减,因为pop()并不是一定执行的
return res;
}
}
提交结果: