栈与队列理论基础
队列是先进先出,栈是先进后出。
用栈实现队列
我们需要用两个栈来实现队列的操作,两个栈方向相反,分别模拟入队列和出队列的效果。难点就在于出队列,因为我们先入栈,先进去的后出来,例如放入了123,所以我们应该先将stackIn中的元素放入到stackOut中去,这样栈底就是3,最先出来就是1,就模拟了出队列的效果。
class MyQueue {
public:
stack<int> stackIn;
stack<int> stackOut;
MyQueue() {
}
void push(int x) {
stackIn.push(x);
}
int pop() {
if(stackOut.empty())
{
while(!stackIn.empty())
{
stackOut.push(stackIn.top());
stackIn.pop();
}
}
int result = stackOut.top();
stackOut.pop();
return result;
}
int peek() {
int result = this->pop();
stackOut.push(result);
return result;
}
bool empty() {
return stackOut.empty() && stackIn.empty();
}
};
注意,这里面peek()函数是直接调用了deque的pop()功能,得到的就是最先出来的元素,因位我们只想知道这个元素是多少,所有我们还需要将这个调出来的元素再放回去。
用队列实现栈
用队列实现栈,其实可以不需要两个队列来实现,我们可以直接使用一个队列,重点在于栈的出战栈,当栈需要出元素时,我们可以将队列的size-1个元素都移到后面,然后拿出队列的第一个元素,这样就实现了出栈的效果。
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();
}
};