day 9
了解了一下栈的api
pop
push
top
empty
本题很简单
思路:
多用一个栈当作工具,方便调整弹出的顺序和队列弹出元素的顺序一致
插入:插入stIn
弹出:先把stIn的元素全部弹出到stOut,再从stOut弹出元素
获取首元素:
1. 先把stIn的元素全部弹出到stOut,再对stOut 进行 top函数即可
2.函数复用:调用之前写的pop函数,将栈顶元素获取之后,再将元素放回
class MyQueue {
stack<int> stIn;
stack<int> stOut;
public:
MyQueue() {
}
void push(int x) {
stIn.push(x);
}
int pop() {
if(stOut.empty())
{
while(!stIn.empty())
{
stOut.push(stIn.top());
stIn.pop();
}
}
int res = stOut.top();
stOut.pop();
return res;
}
int peek() {
int res = this->pop();
stOut.push(res);
return res;
}
bool empty() {
return stOut.empty() && stIn.empty();
}
};