题解 | #用两个栈实现队列#

用两个栈实现队列

https://www.nowcoder.com/practice/54275ddae22f475981afa2244dd448c6

import java.util.*;
import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();
    
    public void push(int node) {
	  // 先将元素压入栈1
        stack1.add(node);
    }
    
    public int pop() {
        if(stack2.isEmpty()){
		  // 栈2为空,就将栈1的元素全部压入栈2
            while(!stack1.isEmpty()){
                stack2.add(stack1.pop());
            }
		  // 然后出栈一个元素
            return stack2.pop();
        }else{
		  // 栈2不为空就出栈一个元素
            return stack2.pop();
        }
    }
}

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务