一、用栈实现队列
思路
用两个栈进行模拟队列,在push数据的时候,将数据放进输入栈,在pop的时候,如果输出栈为空,将输入栈中的所有元素都放入到输出栈里,再从出栈弹出数据,如果输出栈不为空,则直接从出栈弹出数据
代码实现
class MyQueue {
public:
stack<int> stIn;
stack<int> stOut;
MyQueue() {
}
void push(int x) {
stIn.push(x);
}
int pop() {
//只有当stOut为空的时候,再从stIn里导入数据
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(); // 直接使用已有的pop函数
stOut.push(res); // pop函数弹出了元素res,所以在添加回去
return res;
}
bool empty() {
return stIn.empty() && stOut.empty();
}
};
二、用队列实现栈
思路
用一个队列进行模拟,将队列头部的元素(除了最后一个元素外) 重新添加到队列尾部,此时弹出元素顺序就是栈的顺序。
代码实现
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();
}
};