【栈和队列】【打卡第157道】:232. 用栈实现队列

1、题目描述

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

实现 MyQueue 类：

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

 

2、算法分析

首先说一下这些方法是干嘛的。

MyQueue()：初始化定义的栈

push()：向栈中添加数据

pop()：出栈，出栈之前首先将stack1的数据添加到stack2中，然后出栈stack2中的元素

peek()：出栈，和pop的区别就是pop会弹出元素，peek会显示元素，但不会弹栈元素。

3、代码实现

import java.util.*;
class MyQueue {
    Stack<Integer> stack1;
    Stack<Integer> stack2;
    public MyQueue() {
        stack1 = new Stack<>();
        stack2 = new Stack<>();
    }
    
    public void push(int x) {
        stack1.push(x);
    }
    
    public int pop() {
        dump();
        return stack2.pop();
    }
    
    public int peek() {
        dump();
        return stack2.peek();
    }
    
    public boolean empty() {
        return stack1.isEmpty() && stack2.isEmpty();
    }

    public void dump(){
        if(stack2.isEmpty()){
            while(!stack1.isEmpty()){
                stack2.push(stack1.pop());
            }
        }
    }
}

/**
 * Your MyQueue object will be instantiated and called as such:
 * MyQueue obj = new MyQueue();
 * obj.push(x);
 * int param_2 = obj.pop();
 * int param_3 = obj.peek();
 * boolean param_4 = obj.empty();
 */