用两个栈实现队列

题目描述

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。

知识点

栈和队列（先进后出和先进先出）

分析

边界条件：由于输入int型，无边界条件

自己的思路

push操作：若栈2为空，则直接往里push。若不为空，则先把栈2的元素依次出栈到栈1中，把新元素push到栈2，再把栈1的元素依次出栈到栈2中。通过这样的方式，保证每次入栈后新入栈的数据都是在最底层，最老的数据都是在最顶层。

pop操作：直接pop栈2的元素即可。

复杂度

push 时间O(n) ，空间复杂度O(1)

pop  时间O(1),    空间O(1)

代码

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
       while(!stack2.isEmpty()){
           stack1.push(stack2.pop());
       }
        stack2.push(node);
        while(!stack1.isEmpty()){
            stack2.push(stack1.pop());
        }
    }
    
    public int pop() {
        return stack2.pop();
    }
}

优化

看了别人的解法，可以将push和pop的时间复杂度降到O(1)

思路

push操作就直接入栈，不再进行元素的挪动，比如都如栈1；

pop操作需要判断栈2是否为空，若为空，则把栈1的元素依次推入栈2，并出栈顶元素。若不为空，则出栈2 顶部的元素。空了之后又把栈1的转过来，再出栈2的。

总结

关键点：对栈和队列两种数据结构的认识。优化的点在于后一种思路避免了每一次入栈都要挪动栈中的所有元素。