由两个栈组成的队列

最新推荐文章于 2024-01-20 20:46:33 发布

原创最新推荐文章于 2024-01-20 20:46:33 发布 · 172 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#栈和队列 #由两个栈组成的队列

算法同时被 2 个专栏收录

78 篇文章

订阅专栏

java

9 篇文章

订阅专栏

博客主要讲述编写一个类，用两个栈实现队列，支持队列基本操作add、poll、peek。同时给出两个操作规则，一是stackPush往stackPop压入数据时要一次性全压入；二是stackPop不为空时，stackPush不能向其压入数据。

题目
编写一个类，用两个栈实现队列，支持队列的基本操作(add、poll、peek)。

解答
必须做到以下两点：
1. 如果stackPush 要往 stackPop 中压入数据，那么必须一次性把stackPush中的数据全部压入。
2. 如果stackPop 不为空，stackPush绝对不能向stackPop中压入数据。

代码

// 用两个栈组成的队列

import java.util.*;

public class TwoStackQueue {
    public Stack<Integer> stackPush;
    public Stack<Integer> stackPop;

    public TwoStackQueue() {
        stackPush = new Stack<Integer>();
        stackPop = new Stack<Integer>();
    }

    public void add(int pushInt) {
        stackPush.push(pushInt);
    }

    public int poll() {
        if (stackPop.empty() && stackPush.empty()) {
            throw new RuntimeException("Queue is empty!");
        } else if (stackPop.empty()) {
            while (!stackPush.empty()) {
                stackPop.push(stackPush.pop());
            }
        }
        return stackPop.pop();
    }

    public int peek() {
        if (stackPop.empty() && stackPush.empty()) {
            throw new RuntimeException("Queue is empty!");
        } else if (stackPop.empty()) {
            while (!stackPush.empty()) {
                stackPop.push(stackPush.pop());
            }
        }
        return stackPop.peek();
    }

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