day10 - 栈与队列 part01
力扣232. 用栈实现队列
思路:建立一个输入栈,一个输出栈,每次push进队列时,放进输入栈,出队列时,从输出栈弹出,如果输出栈是空的,就把输入栈的全部数都弹出到输出栈。
代码
class MyQueue {
public:
MyQueue() {
}
void push(int x) {
stIn.push(x);
}
int pop() {
if (stOut.empty())
{
while (!stIn.empty())
{
stOut.push(stIn.top());
stIn.pop();
}
}
int result = stOut.top();
stOut.pop();
return result;
}
int peek() {
int res = this->pop();
stOut.push(res);
return res;
}
bool empty() {
return stIn.empty() && stOut.empty();
}
private:
stack<int> stIn;
stack<int> stOut;
};
力扣225. 用队列实现栈
思路:用一个队列,然后每次进队列正常进,并记录size,出栈时候,把队列中size-1个全部弹出并重新加入队列,返回最后一个。
代码
lass MyStack {
public:
MyStack() {
size = 0;
}
void push(int x) {
que.push(x);
size++;
}
int pop() {
for (int i = 0; i < size - 1; i++)
{
int tmp = que.front();
que.pop();
que.push(tmp);
}
int result = que.front();
que.pop();
size--;
return result;
}
int top() {
return que.back();
}
bool empty() {
return size == 0;
}
private:
queue<int> que;
int size;
};