Leetcode 225.用队列实现栈
问题描述:
使用队列实现栈的下列操作:
push(x) -- 元素 x 入栈
pop() -- 移除栈顶元素
top() -- 获取栈顶元素
empty() -- 返回栈是否为空
注意:
你只能使用队列的基本操作-- 也就是 push to back, peek/pop from front, size, 和 is empty 这些操作是合法的。
你所使用的语言也许不支持队列。 你可以使用 list 或者 deque(双端队列)来模拟一个队列 , 只要是标准的队列操作即可。
你可以假设所有操作都是有效的(例如, 对一个空的栈不会调用 pop 或者 top 操作)。
问题思路:
采用两个队列,一个为辅助队列。将第一个队列a中元素弹出只剩一个,辅助队列b存储a中弹出的元素。取走a中剩下的一个元素,又将b与a进行交换,重复操作直到两个队列为空。
class MyStack {
public:
queue<int>a, b;
/** Initialize your data structure here. */
MyStack() {
}
/** Push element x onto stack. */
void push(int x) {
a.push(x);
}
/** Removes the element on top of the stack and returns that element. */
int pop() {
if (!empty())
{
int t = top();
a.pop();
return t;
}
return -1;
}
/** Get the top element. */
int top() {
if (!empty())
{
if (a.empty())
a.swap(b);
while (a.size() != 1)
{
int t = a.front();
b.push(t);
a.pop();
}
return a.front();
}
return -1;
}
/** Returns whether the stack is empty. */
bool empty() {
return a.empty() && b.empty();
}
};