请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):(来源力扣)
实现 MyQueue 类:
void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空,返回 true ;否则,返回 false
说明:
你只能使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
class MyQueue {
//使用栈来模拟队列,需要两个栈,一个为输入栈,一个为输出栈
Stack<Integer> stack1=new Stack<>();//输入栈
Stack<Integer> stack2=new Stack<>();//输出栈
/** Initialize your data structure here. */
public MyQueue() {
}
/** Push element x to the back of queue. */
//添加元素添加到输入栈即可
public void push(int x) {
stack1.push(x);
}
/** Removes the element from in front of queue and returns that element. */
//输出栈栈顶元素就是最早加入的元素,如果不为空,直接pop(),若为空,将输入栈中的元素弹栈压栈到输出栈,然后pop()
public int pop() {
if(stack2.empty()){
while(!stack1.empty()){
stack2.push(stack1.pop());
}
}
//int num=stack2.peek();//如果需要使用栈顶元素做判断之类的可以使用peek()方法
//stack2.pop();
//return num;
return stack2.pop();
}
/** Get the front element. */
//队列头就是输出栈的栈顶元素,若输出栈为空,就把输入栈导入到输出栈
public int peek() {
if(!stack2.empty()){
return stack2.peek();
}else{
while(!stack1.empty()){
stack2.push(stack1.pop());
}
return stack2.peek();
}
}
/** Returns whether the queue is empty. */
//队列为空就是输出栈和输入栈均为空的情况
public boolean empty() {
return stack1.empty()&&stack2.empty();
}
}