剑指Offer(牛客网) 用两个栈来实现队列

题目描述:
用两个栈来实现一个队列,完成队列的Push和Pop操作。 队列中的元素为int类型

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
        stack1.push(node);
    }
    
    public int pop() {
    //当弹出时,当 stack2 不为空,弹出 stack2 栈顶元素,
    //如果 stack2 为空,将 stack1 中的全部数逐个出栈入栈 stack2,
    //再弹出 stack2 栈顶元素
            if (stack2.size() == 0) {
            while (stack1.size() > 0) {
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}
发布了24 篇原创文章 · 获赞 2 · 访问量 367

猜你喜欢

转载自blog.csdn.net/Iloveyshsh/article/details/105333565
今日推荐