LeetCode232.用栈实现队列

发布时间 2023-10-30 19:34:42作者: 白布鸽

题目描述

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):
实现 MyQueue 类:
void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空,返回 true ;否则,返回 false

示例

image

提交的代码

import java.util.Stack;
class MyQueue {

    Stack<Integer> inStack;
    Stack<Integer> outStack;
    
    public MyQueue() {
        inStack=new Stack<>();
        outStack=new Stack<>();
    }
    
    public void push(int x) {
        inStack.push(x);
    }
    
    public int pop() {
        dumpStack();
        return outStack.pop();
    }
    
    public int peek() {
        dumpStack();
        return outStack.peek();
    }
    
    public boolean empty() {
        return inStack.isEmpty()&&outStack.isEmpty();
    }

    public void dumpStack(){
        if(!outStack.isEmpty())return;
        while(!inStack.isEmpty()){
            outStack.push(inStack.pop());
        }
    }
}

实现方法比较简单,push操作全部放入inStack,如果是pop的话,先判断outStack是否为空,如果为空则将InStack内的元素倒灌入outStack中,然后pop出outStack栈顶元素;如果outStack不为空,就不能再倒灌inStack进outStack中了,会破坏outStack的结构,先优先pop出outStack中的内容。peek类似pop。