栈与队列|232.用栈实现队列 、225. 用队列实现栈
232.用栈实现队列
自己审题思路
使用两个栈实现队列,将一个栈的元素依次弹进另一个栈,然后再从另一个栈依次弹出
看完代码随想录题解后的收获
思路基本一致
代码:
class MyQueue {
public:
stack<int> stIn;
stack<int> stOut;
/** Initialize your data structure here. */
MyQueue() {
}
/** Push element x to the back of queue. */
void push(int x) {
stIn.push(x);
}
/** Removes the element from in front of queue and returns that element. */
int pop() {
// 只有当stOut为空的时候,再从stIn里导入数据(导入stIn全部数据)
if (stOut.empty()) {
// 从stIn导入数据直到stIn为空
while(!stIn.empty()) {
stOut.push(stIn.top());
stIn.pop();
}
}
int result = stOut.top();
stOut.pop();
return result;
}
/** Get the front element. */
int peek() {
int res = this->pop(); // 直接使用已有的pop函数
stOut.push(res); // 因为pop函数弹出了元素res,所以再添加回去
return res;
}
/** Returns whether the queue is empty. */
bool empty() {
return stIn.empty() && stOut.empty();
}
};
在实现
pop()
函数时,没有写if (stOut.empty())
导致错误,当stOut还没有清空前,如果将stIn元素弹入,会改变元素的次序导致错误。
参考详解
225. 用队列实现栈
自己审题思路
使用两个队列实现栈,但是怎么改变次序一时没想到
看完代码随想录题解后的收获
利用que2来备份前n -1个元素,将que1最后一个元素留下弹出,之后再将que2导回que1。
代码:
class MyStack {
public:
queue<int> que1;
queue<int> que2;
MyStack() {
}
void push(int x) {
que1.push(x);
}
int pop() {
int size = que1.size() - 1;
while(size--) {
int ans = que1.front();
que1.pop();
que2.push(ans);
}
int result = que1.front();
que1.pop();
que1 = que2;
while(!que2.empty()) {
que2.pop();
}
/*
while(!que2.empty()) { // 也可以这样实现 que1 = que2;
int ans = que2.front();
que2.pop();
que1.push(ans);
} */
return result;
}
int top() {
return que1.back();
}
bool empty() {
return que1.empty();
}
};
/**
* 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();
*/
参考详解
今日感悟
栈和队列的基本原理要理解,用队列实现栈卡了下,需要加深理解。