参考自:《剑指Offer——名企面试官精讲典型编程题》
题目:用两个栈实现队列
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别完成在队列尾部插入节点和在队列头部删除节点的功能。
主要思路:结合栈先进后出和队列先进先出的特点。先把元素压入第一个栈,再把元素弹出到第二个栈,使得最开始插入的元素翻转
到第二个栈的顶部。在第一个栈上执行插入操作;在第二个栈上执行删除操作,若第二个栈为空,则把第一个栈的元素弹出到到第二个栈。
关键点:栈和队列的特点
时间复杂度:插入操作O(1);删除操作O(n)
public class SimulateQueue {
/**
* 输入栈
*/
private Stack<Integer> inputStack = new Stack<>();
/**
* 输出栈
*/
private Stack<Integer> outputStack = new Stack<>();
public static void main(String[] args) {
SimulateQueue simulateQueue = new SimulateQueue();
simulateQueue.appendTail(1);
simulateQueue.appendTail(2);
simulateQueue.appendTail(3);
while (!simulateQueue.isEmpty()) {
System.out.println(simulateQueue.deleteHead());
}
}
public boolean isEmpty() {
return inputStack.isEmpty() && outputStack.isEmpty();
}
public void appendTail(int node) {
inputStack.push(node);
}
public int deleteHead() {
//输出栈为空,则把输入栈的元素弹出到输出栈
if (outputStack.isEmpty()) {
while (!inputStack.isEmpty()) {
outputStack.push(inputStack.pop());
}
}
if (outputStack.isEmpty()) {
return -1;
}
return outputStack.pop();
}
}