leetcode用栈实现队列

leetcode 地址

https://leetcode.cn/problems/implement-queue-using-stacks/

题目描述

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列支持的所有操作（push、pop、peek、empty）：

实现 MyQueue 类：

void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空，返回 true ；否则，返回 false

重点：

1. 每次push，push到push栈中，pop都从pop栈中pop
2. 只有pop栈中为空的时候再从，push栈中导入全部数据

参考代码

class MyQueue {
public:
    stack<int> _push_stack;
    stack<int> _pop_stack;
    MyQueue() {

    }
    
    void push(int x) {
        _push_stack.push(x);
    }
    
    int pop() {
        //每次弹出都是从pop栈中弹出，只有pop栈中为空的时候才能从push栈中导入
        if(_pop_stack.empty()) {
            while(!_push_stack.empty()) {
                int temp = _push_stack.top();
                _pop_stack.push(temp);
                _push_stack.pop();
            }
        }

        if(!_pop_stack.empty()) {
            int result = _pop_stack.top();
            _pop_stack.pop();
            return result;
        }
        return -1;
    }
    
    int peek() {
        if(!this->empty()) {
            //因为每次都是从pop栈中弹出元素，所以需要将元素再push进pop栈中
            int result = this->pop();
            _pop_stack.push(result);
            return result;
        }
        return -1;
    }
    
    bool empty() {
        if(_pop_stack.empty() && _push_stack.empty()) {
            return true;
        } else {
            return false;
        }
    }
};