【简单】面试题 03.04. 化栈为队-优快云博客

本文介绍了一种使用两个栈来实现队列的方法，通过MyQueue类的定义，演示了如何进行元素的入队(push)、出队(pop)、查看队首(peek)以及判断队列是否为空(empty)的操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

【题目】
实现一个MyQueue类，该类用两个栈来实现一个队列。
来源：leetcode
链接：https://leetcode-cn.com/problems/implement-queue-using-stacks-lcci/
【示例】
MyQueue queue = new MyQueue();
queue.push(1);
queue.push(2);
queue.peek(); // 返回 1
queue.pop(); // 返回 1
queue.empty(); // 返回 false
【提示】
1、你只能使用标准的栈操作 – 也就是只有 push to top, peek/pop from top, size 和 is empty 操作是合法的。
2、你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。
3、假设所有操作都是有效的（例如，一个空的队列不会调用 pop 或者 peek 操作）
【代码】

class MyQueue {
public:
    /** Initialize your data structure here. */
    stack<int> s1;
    stack<int> s2;
    MyQueue() {
    }
    /** Push element x to the back of queue. */
    void push(int x) {
        s1.push(x);
    }
    /** Removes the element from in front of queue and returns that element. */
    int pop() {
        int x=-1;
        if(s2.size()==0){
            while(!s1.empty()){
                x=s1.top();
                s1.pop();
                s2.push(x);
            }            
        }
        x=s2.top();
        s2.pop();
        return x;
    }
    /** Get the front element. */
    int peek() {
        int x=-1;
        if(s2.size()==0){
            while(!s1.empty()){
                x=s1.top();
                s1.pop();
                s2.push(x);
            }            
        }
        x=s2.top();
        return x;
    }
    /** Returns whether the queue is empty. */
    bool empty() {
        if(!s1.size()&&!s2.size())
            return true;
        return false;
    }
};
/**
 * Your MyQueue object will be instantiated and called as such:
 * MyQueue* obj = new MyQueue();
 * obj->push(x);
 * int param_2 = obj->pop();
 * int param_3 = obj->peek();
 * bool param_4 = obj->empty();
 */