LeetCode232.用栈实现队列

题目描述

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

示例

提交的代码

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。