题目:用队列实现栈
题目描述:
使用队列实现栈的下列操作:
push(x) – 元素 x 入栈
pop() – 移除栈顶元素
top() – 获取栈顶元素
empty() – 返回栈是否为空
测试用例:
输入:[“MyStack”,“push”,“push”,“top”,“pop”,“empty”] [[],[1],[2],[],[],[]]
输出:[null,null,null,2,2,false]
代码实现如下:
class MyStack {
public:
/** Initialize your data structure here. */
MyStack() {
}
/** Push element x onto stack. */
void push(int x) {
//队列q1不为空,压到q1中
if(!q1.empty()){
q1.push(x);
}
else{
q2.push(x);
}
}
/** Removes the element on top of the stack and returns that element. */
int pop() {
int res;
if(q1.empty()){
//将size() - 1个数字挪到q1中
while(q2.size() > 1){
int p = q2.front();
q2.pop();
q1.push(p);
}
//取出q2的最后一个元素,则是栈顶元素,删除
res = q2.front();
q2.pop();
}
else{
if(q2.empty()){
while(q1.size() > 1){
int p = q1.front();
q1.pop();
q2.push(p);
}
res = q1.front();
q1.pop();
}
}
return res;
}
/** Get the top element. */
int top() {
int ret;
if(!q1.empty()){
ret = q1.back();
}
else{
if(!q2.empty()){
ret = q2.back();
}
}
return ret;
}
/** Returns whether the stack is empty. */
bool empty() {
return q1.empty() && q2.empty();
}
private:
std::queue<int> q1;
std::queue<int> q2;
};
/**
* Your MyStack object will be instantiated and called as such:
* MyStack* obj = new MyStack();
* obj->push(x);
* int param_2 = obj->pop();
* int param_3 = obj->top();
* bool param_4 = obj->empty();
*/