用两个栈实现一个队列。
队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,
分别完成在队列尾部插入整数和在队列头部删除整数的功能。
(若队列中没有元素,deleteHead 操作返回 -1 )
示例 1:
输入: ["CQueue","appendTail","deleteHead","deleteHead"] [[],[3],[],[]] 输出:[null,null,3,-1]
提示:
1 <= values <= 10000
最多会对 appendTail、deleteHead 进行 10000 次调用
题解:
使用两个栈stack1,stack2,其中stack1负责插入,stack2负责删除。
删除时,如果stack2为空,则先把stack1的数据出到stack2中,
然后再stack2出栈,如果此时stack2仍为空,则返回-1.
public class offer9 {
public static void main(String[] args) {
// TODO Auto-generated method stub
CQueue obj = new CQueue();
obj.appendTail(3);
int param = obj.deleteHead();
int param_1 = obj.deleteHead();
System.out.println(param_1);
}
}
class CQueue{
private Stack stack1;
private Stack stack2;
public CQueue(){
stack1 = new Stack();
stack2 = new Stack();
}
public void appendTail(int value){
stack1.add(value);
}
public int deleteHead(){
if(stack2.isEmpty()){
while(!stack1.isEmpty()){
stack2.push(stack1.pop());
}
}
return stack2.isEmpty()?-1:(int) stack2.pop();
}
}