题目来源
题目概述
请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(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(双端队列)来模拟一个栈,只要是标准的栈操作即可。
思路分析
在昨天的题目《力扣225用队列实现栈》中使用了在push的时候进行反转的做法,现在就使用在插入的时候进行反转。
代码实现
java实现
public class MyQueue {
Stack<Integer> stack = new Stack<>();
Stack<Integer> temp = new Stack<>();
public MyQueue() {}
public void push(int x) {
int size = stack.size();
for (int i = 0; i < size; i++) {
temp.push(stack.pop());
}
stack.push(x);
for (int i = 0; i < size; i++) {
stack.push(temp.pop());
}
}
public int pop() {
return stack.pop();
}
public int peek() {
return stack.peek();
}
public boolean empty() {
return stack.isEmpty();
}
}
c++实现
class MyQueue {
private:
stack<int> s;
stack<int> temp;
public:
MyQueue() {}
void push(int x) {
int size = s.size();
for (int i = 0; i < size; i++) {
temp.push(s.top());
s.pop();
}
s.push(x);
for (int i = 0; i < size; i++) {
s.push(temp.top());
temp.pop();
}
}
int pop() {
int res = s.top();
s.pop();
return res;
}
int peek() {
return s.top();
}
bool empty() {
return s.empty();
}
};