参考自:《剑指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();
}
}