LeetCode 232. 用栈实现队列
解题思路
首先明确队列和栈的区别:栈是先进后出、队列是先进先出。要使用栈来模拟队列的行为,需要用到两个栈: stack_in、 stack_out。
push数据的时候,只需要把数据放进输入栈即可。但在pop的时候,操作就复杂一些。如果输出栈为空,就需要把输入栈里面的数据全部导入进来,然后从出栈弹出数据。如果输出栈不为空,则直接从出栈弹出数据即可。
代码解析
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();
}
};
LeetCode 232. 用栈实现队列
解题思路
一个队列在模拟栈弹出元素的时候只要将队列头部的元素(除了最后一个元素外) 重新添加到队列尾部,此时再去弹出元素就是栈的顺序了。
代码解析
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 result = que.front();
que.pop();
return result;
}
int top() {
return que.back();
}
bool empty() {
return que.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();
*/
容器常见的使用方法
stack
stack<int> st;
// 添加元素
st.push(10);
// 获取栈顶元素
st.top();
// 移除栈顶元素
st.pop();
queue
#include <iostream>
#include <queue>
using namespace std;
int main() {
queue<int> q;
q.push(10); // Enqueue 10
q.push(20); // Enqueue 20
q.push(30); // Enqueue 30
cout << q.front() << endl; // Peek 10
cout << q.back() <<endl;
cout << q.size() << endl;
q.pop(); // Dequeue 10
cout << q.front() << endl; // Peek 20
return 0;
}