Java两个栈实现一个队列

Java:两个栈实现一个队列

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();

    /*
     * stack1和stack2所使用的方法，均为Java提供的Stack类中自带的方法，而不是递归调用，二者只是名字相同
     * */

    //进栈
    public void push(int node) {
        //如果进栈次数多，出栈次数少，则stack2中存有元素，故需要先判断stack2中如果不为空，先把stack2中的元素转移到stack1中
        while (!stack2.empty()) {
            stack1.push(stack2.pop());
        }

        stack1.push(node);//把元素进stack1中

    }

    //出栈
    public int pop() {
        //把stack1中的元素全部转移到stack2中
        while (!stack1.empty()) {
            stack2.push(stack1.pop());
        }

        return stack2.pop();//返回stack2出栈的结果
    }


}

测试代码

import java.util.Random;

public class demomain {
    public static void main(String[] args) {
        Solution s = new Solution();
        Random r = new Random();

        for (int i = 0; i < 5; i++) {
            int data = r.nextInt(100);
            System.out.println(data + " 进队");
            s.push(data);   //生成5个随机数，并进栈
        }
        System.out.println("============================");
        System.out.println(s.pop() + " 出队");  //调用一次pop()方法，把stack1中的元素全部转移到stack2中，并输出栈顶元素
        while (!s.stack2.empty()) {
            System.out.println(s.pop() + " 出队");//遍历stack2中剩余元素并输出
        }


    }
}

这篇文章参考了下面文章和视频

文章

视频