力扣232. 用栈实现队列:
使用两个栈实现队列
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):
实现 MyQueue 类:
void push(int x)
将元素 x 推到队列的末尾int peek()
将元素 x 推到队列的末尾int pop()
将元素 x 推到队列的末尾boolean empty()
将元素 x 推到队列的末尾说明:
说明:
- 你只能使用标准的栈操作 —— 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
- 你所使用的语言也许不支持栈。你可以使用 list 或者 deque(双端队列)来模拟一个栈,只要是标准的栈操作即可。
问题描述
提示:事实只需要实现peek()函数,其他函数基本保持不变
题目原题给出如下代码:
class MyQueue {
public MyQueue() {
}
public void push(int x) {
}
public int pop() {
}
public int peek() {
}
public boolean empty() {
}
}
/**
* 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();
*/
题解分析:
分析:使用栈实现队列,只需要使用两个栈,一个栈负责入队列,即实现push函数,一个栈负责出队列,即实现pop函数
具体实现如下:stack1
负责接受元素入队,而当需要使用peek
查看队首元素时,先判断stack2
是否为空,如果为空,则需要将stack1
中的元素全部弹出栈放入stack2
中,否则直接返回,这样的话stack2
的栈顶端则为需要查看的元素,同理也为需要弹出的元素,即pop()
函数
解决方案:
提示:下面为题解答案,先在构造器中创建两个stack对象,push函数直接调用即可,pop函数则需要将stk1的元素全部放入stk2中,这段代码在peek函数中实现,避免冗余,判空函数则需要两个栈都为空时队列才空
class MyQueue {
Stack<Integer> stk1,stk2;
public MyQueue() {
stk1=new Stack<>();
stk2=new Stack<>();
}
public void push(int x) {
stk1.push(x);
}
public int pop() {
int popVal=peek();
stk2.pop();
return popVal;
}
public int peek() {
if(stk2.isEmpty()){
while(!stk1.isEmpty()){
stk2.push(stk1.pop());
}
}
return stk2.peek();
}
public boolean empty() {
return stk1.isEmpty()&&stk2.isEmpty();
}
}