题目描述:
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 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]
解题思路:
队列的特点是先进先出,而栈的特点是先进后出,两种数据结构的特点正好相反,因此可以使用两个栈来实现队列,因为第一次出栈的顺序和队列出队的顺序是相反的,如果在某一个元素第一次出栈的时候再次将该元素入第二个栈的话,那么该元素再次出栈后的顺序就和在队列中出队的顺序是一致的了。
实现的细节:有两个in
和out
栈来处理入队和出队的操作,当有元素入队的时候,直接将该元素放入in
栈中;当有元素出队的时候,判断out
栈中有没有元素,有两种情况:(1)out
栈没有元素的话,则将in
栈中所有元素出栈并弹入out
栈中,然后弹出out
栈中的元素(这样就可以保证先进先出的顺序);如果in
栈中也没有元素则表示队列中没有元素了;(2)out
栈中有元素则直接弹出元素。
实现代码:
Stack<Integer> in = new Stack<Integer>();
Stack<Integer> out = new Stack<Integer>();
public void appendTail(int value) {
in.push(value);
}
public int deleteHead() {
if(out.isEmpty()){
while(!in.isEmpty())//将in栈中的元素移动out栈中,这样经过两次反转后就是先进先出的顺序了
out.push(in.pop());
}
if(out.isEmpty())//说明队列中没有元素
return -1;
return out.pop();
}