【队列】LeetCode 232. 用栈实现队列

题目链接

232. 用栈实现队列

思路

设置一个主栈 mainStack 和一个辅助栈 assistantStack，在进行入队的时候，将 mainStack 中的元素全部放入 assistantStack 中，再将 x 入队，然后再将 assistantStack 的元素放入 mainStack 中。

代码

class MyQueue {

    Stack<Integer> mainStack;
    Stack<Integer> assistantStack;
    public MyQueue() {
        mainStack = new Stack<>();
        assistantStack = new Stack<>();
    }

    public void push(int x) {
        if(mainStack.empty()){
            mainStack.push(x);
        }else{
            while(!mainStack.empty()){
                assistantStack.push(mainStack.pop());
            }

            mainStack.push(x);
            while(!assistantStack.empty()){
                mainStack.push(assistantStack.pop());
            }
        }
    }

    public int pop() {
        return mainStack.pop();
    }

    public int peek() {
        return mainStack.peek();
    }

    public boolean empty() {
        return mainStack.empty();
    }
}

拓展

【队列】LeetCode 225. 用队列实现栈