剑指Offer_简单_JZ5_用两个栈实现队列

流浪所

于 2021-03-29 11:41:42 发布

阅读量69

点赞数

分类专栏：算法文章标签：算法

本文链接：https://blog.csdn.net/m0_49068745/article/details/115297546

版权

算法专栏收录该内容

6 篇文章 0 订阅

订阅专栏

题目描述

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

分析

确定stack1为入栈，stack2为出栈（即入数据永远都往stack1中入，出数据永远都从stack2中出，但是当stack2为空时，出数据得从stack1中倒数据），因为将栈中的元素倒一遍就将数据倒序了，就可以达到队列的效果。

解题思路

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(stack1.empty()&&stack2.empty())
          throw new RuntimeException("Queue is empty!");
        if(stack2.empty()){
            while(!stack1.empty()){
                stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}