仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(push、pop、peek、empty):
实现 MyQueue 类:
- void push(int x) 将元素 x 推到队列的末尾
- int pop() 从队列的开头移除并返回元素
- int peek() 返回队列开头的元素
- boolean empty() 如果队列为空,返回 true ;否则,返回 false
使用栈实现队列需要两个栈,一个输入栈,一个输出栈。
1.在输入元素时只需要把元素放入输入栈。
2.在输出时需要判断输出栈是否为空,如果为空,就要把输入栈中的元素全部导入,再从输出栈弹出数据;如果不为空,可以直接弹出元素。
3.若进栈和出栈均为空,说明模拟的队列空了。
实现代码如下:
stack<int> stIn; //入栈
stack<int> stOut; //出栈
MyQueue() {}
//将元素x入队
void push(int x) {
stIn.push(x);
}
//从队列首部移除元素
int pop() {
if(stOut.empty()) {
while(! stIn.empty()) {
int i = stIn.top();
stOut.push(i);
stIn.pop();
}
}
int r = stOut.top();
stOut.pop();
return r;
}
//返回队首元素
int peek() {
int res = this->pop();
stOut.push(res);
return res;
}
//判断队列是否为空
bool empty() {
return stIn.empty() && stOut.empty();
}