剑指Offer-题9(Java版):用两个栈实现队列

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

猜你喜欢

转载自blog.csdn.net/m0_37862405/article/details/80075288