代码随想录算法训练营第十天 | 232.用栈实现队列 225.用队列实现栈
用栈实现队列
思路:
在push数据的时候,只要数据放进输入栈就好,但在pop的时候,操作就复杂一些,输出栈如果为空,就把进栈数据全部导入进来(注意是全部导入),再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据就可以了。
最后如何判断队列为空呢?如果进栈和出栈都为空的话,说明模拟的队列为空了。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34
| class MyQueue { public: stack<int> stIn; stack<int> stOut; 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(); } };
|
用队列实现栈
思路:
队列是先进先出的规则,把一个队列中的数据导入另一个队列中,数据的顺序并没有变,并没有变成先进后出的顺序。
所以用栈实现队列, 和用队列实现栈的思路还是不一样的,这取决于这两个数据结构的性质。
但是依然还是要用两个队列来模拟栈,只不过没有输入和输出的关系,而是另一个队列完全用来备份的!
如下面动画所示,用两个队列que1和que2实现队列的功能,que2其实完全就是一个备份的作用,把que1最后面的元素以外的元素都备份到que2,然后弹出最后面的元素,再把其他元素从que2导回que1。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
| class MyStack { public: queue<int> que; MyStack() {
}
void push(int x) { que.push(x); }
int pop() { int size = que.size(); size--; while (size--) { que.push(que.front()); que.pop(); } int res = que.front(); que.pop(); return res; }
int top() { return que.back(); }
bool empty() { return que.empty(); } };
|