0
点赞
收藏
分享

微信扫一扫

用栈实现队列(双栈,输入栈、输出栈)

闲鱼不咸_99f1 2022-04-20 阅读 82
java

题目:请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):

实现 MyQueue 类:

void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空,返回 true ;否则,返回 false。

解题思路:将一个栈当作输入栈,用于push 传入的数据;另一个栈当作输出栈,用于pop 和peek 操作。

每次pop或peek时,若输出栈为空则将输入栈的全部数据依次push进输出栈,这样输出栈从栈顶往栈底的顺序就是队列从队首往队尾的顺序(实现反转)。

代码如下:

class MyQueue {
    Deque<Integer> inStack;
    Deque<Integer> outStack;

    public MyQueue() {
        inStack = new ArrayDeque<Integer>();
        outStack = new ArrayDeque<Integer>();
    }

    public void push(int x) {
        inStack.push(x);
    }

    public int pop() {
        if (outStack.isEmpty()) {
            outStack_isEmpty();//将输入栈的全部数据依次push进输出栈
        }
        return outStack.pop();
    }

    public int peek() {
        if (outStack.isEmpty()) {//peek和pop同理
            outStack_isEmpty();
        }
        return outStack.peek();
    }

    public boolean empty() {//instack和outstack都有队列数据
        return inStack.isEmpty() && outStack.isEmpty();
    }

    private void outStack_isEmpty() {
        while (!inStack.isEmpty()) {
            outStack.push(inStack.pop());
        }
    }
}
举报

相关推荐

0 条评论