Java语言用两个栈设计一个队列

最新推荐文章于 2022-10-03 11:53:47 发布

原创最新推荐文章于 2022-10-03 11:53:47 发布 · 168 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#JAVA #JAVA 栈运用 #JAVA 两个栈构成队列 #JAVA 栈 #队列

JAVA 专栏收录该内容

29 篇文章

订阅专栏

该博客讨论了如何利用两个栈来设计一个Java队列，详细解释了如何通过这两个栈来实现队列的添加（add）、移除（poll）和查看队首元素（peek）等基本操作。

题目：编写一个类，用两个栈实现队列，支持队列的基本操作（add,poll,peek）

解题：

public class TwoStacksImplementQueue {

        public static class myQueue {

            Stack<Integer> stack1;
            Stack<Integer> stack2;

            public myQueue() {
                stack1 = new Stack<Integer>();
                stack2 = new Stack<Integer>();
            }

            /**
             * add只负责往stack1里面添加数据
             *
             * @param newNum
             */
            public void add(Integer newNum) {
                stack1.push(newNum);
            }

            /**
             * 这里要注意两点：
             * 1.stack1要一次性压入stack2
             * 2.stack2不为空，stack1绝不能向stack2压入数据
             *
             * @return
             */
            public Integer poll() {
                if (stack1.isEmpty() && stack2.isEmpty()) {
                    throw new RuntimeException("Queue is Empty");
                } else if (stack2.isEmpty()) {
                    while (!stack1.isEmpty()) {
                        stack2.push(stack1.pop());
                    }
                }
                return stack2.pop();
            }

            public Integer peek() {
                if (stack1.isEmpty() && stack2.isEmpty()) {
                    throw new RuntimeException("Queue is Empty");
                } else if (stack2.isEmpty()) {
                    while (!stack1.isEmpty()) {
                        stack2.push(stack1.pop());
                    }
                }
                return stack2.peek();
            }
        }

        public static void main(String[] args) {
            myQueue mQueue = new myQueue();
            mQueue.add(1);
            mQueue.add(2);
            mQueue.add(3);
            System.out.println(mQueue.peek());
            System.out.println(mQueue.poll());
            System.out.println(mQueue.peek());
            System.out.println(mQueue.poll());
            System.out.println(mQueue.peek());
            System.out.println(mQueue.poll());
        }
    }