剑指offer——用两个栈实现队列

题目

用两个栈来实现一个队列，完成队列的Push和Pop操作。 队列中的元素为int类型。剑指offer面试题9

代码

import java.util.Stack;

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

思路

1. 队列的特点：先进先出；栈的特点：先进后出
2. 解决问题：如何让先进栈的元素先出来。一开始将push进来的元素存在在stack1中，然后在pop的时候（此时stack2是空栈），把stack1中的元素全部pop到stack2。最先进stack1中的元素就成了stack2中最后进的元素，pop出来就好了。
3. 如果此时还要往队列中push元素，push到stack1中还是stack2中。例如说：一开始stack1中连着push3个数：【1，2，3】。然后pop，那么全部进入到stack2中【3，2，1】,pop掉1，stack2中剩【3，2】。现在要push【4】。如果push到stack1中，不会对2有影响，pop的时候仍然pop stack2中的数据就好；如果push到stack2中，stack2变为【3，2，4】，此时再要pop，应该是【2】，既不能直接pop stack2，也不能全部pop到stack1中。
4. 换句话理解：stack2中是已经整理好pop顺序的元素，有新元素进来，不要去影响它。当stack2中空了以后再放进来。