class MyQueue {//创建集合
Stack<Integer> stackIn;// 用于进队列的栈
Stack<Integer> stackOut;// 用于出队列的栈
public MyQueue() {//初始化数据结构
stackIn = new Stack<>(); // 负责进栈
stackOut = new Stack<>(); // 负责出栈
}
public void push(int x) {
stackIn.push(x);// 直接将元素推入进队列栈
}
// 从队列的前端删除元素并返回该元素
public int pop() {
dumpStackIn(); // 调用 dumpStackIn 函数将元素从进队列栈移到出队列栈
return stackOut.pop(); // 从出队列栈中弹出元素,实现出队列操作
}
//获取队列的前端元素
public int peek() {
dumpStackIn(); // 调用 dumpStackIn 函数将元素从进队列栈移到出队列栈
return stackOut.peek(); // 获取出队列栈的栈顶元素,实现获取队列前端元素的操作
}
public boolean empty() {
return stackIn.isEmpty() && stackOut.isEmpty(); // 只有当两个栈都为空时,队列为空
}
// 如果 stackOut 为空,将 stackIn 中的元素全部移到 stackOut 中
private void dumpStackIn(){
if (!stackOut.isEmpty()) return; // 如果出队列栈不为空,不进行操作
while (!stackIn.isEmpty()){
stackOut.push(stackIn.pop()); // 将进队列栈中的元素逐个移到出队列栈
}
}
}
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push
、pop
、peek
、empty
):
实现 MyQueue
类:
void push(int x)
将元素 x 推到队列的末尾int pop()
从队列的开头移除并返回元素int peek()
返回队列开头的元素boolean empty()
如果队列为空,返回true
;否则,返回false
在这个自定义队列 MyQueue
中,stackIn
是一个 Stack
对象,而 Stack
类是 Java 中的标准库,它提供了用于堆栈操作的方法,包括 push
用于将元素推入堆栈。因此,stackIn.push(x)
表示将元素 x
推入 stackIn
这个堆栈中。
设2个栈,由于栈是先进后出,所以out栈的出栈排列顺序与in栈的出栈排列顺序相反,且out栈与队列的出栈排列顺序相同