用栈实现队列

仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作(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();
}

版权声明:本文为qq_43697646原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。