用两个栈实现队列

请用栈实现一个队列,支持如下四种操作:

  • push(x) – 将元素x插到队尾;
  • pop() – 将队首的元素弹出,并返回该元素;
  • peek() – 返回队首元素;
  • empty() – 返回队列是否为空;
    1
    2
    3
    4
    5
    6
    7
    8
    9
    10
    11
    12
    13
    14
    15
    16
    17
    18
    19
    20
    21
    22
    23
    24
    25
    26
    27
    28
    29
    30
    31
    32
    33
    34
    35
    36
    37
    38
    39
    40
    41
    42
    43
    44
    45
    46
    47
    48
    49
    50
    51
    52
    53
    54
    55
    56
    class MyQueue {
    public:
        /** Initialize your data structure here. */
        stack<int> a, b;
         
        MyQueue() {
             
        }
         
        /** Push element x to the back of queue. */
        void push(int x) {
            a.push(x);
        }
         
        /** Removes the element from in front of queue and returns that element. */
        int pop() {
            while (a.size() > 1) {
                b.push(a.top());
                a.pop();
            }
            int t = a.top(); a.pop();
            while (b.size()) {
                a.push(b.top());
                b.pop();
            }
            return t;
        }
         
        /** Get the front element. */
        int peek() {
            while (a.size() > 1) {
                b.push(a.top());
                a.pop();
            }
            int t = a.top();
            while (b.size()) {
                a.push(b.top());
                b.pop();
            }
            return t;
        }
         
        /** Returns whether the queue is empty. */
        bool empty() {
            return a.empty();
        }
    };
     
    /**
     * Your MyQueue object will be instantiated and called as such:
     * MyQueue obj = MyQueue();
     * obj.push(x);
     * int param_2 = obj.pop();
     * int param_3 = obj.peek();
     * bool param_4 = obj.empty();
     */

      

posted @   !&&||  阅读(16)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· 被坑几百块钱后,我竟然真的恢复了删除的微信聊天记录!
· 没有Manus邀请码?试试免邀请码的MGX或者开源的OpenManus吧
· 【自荐】一款简洁、开源的在线白板工具 Drawnix
· 园子的第一款AI主题卫衣上架——"HELLO! HOW CAN I ASSIST YOU TODAY
· Docker 太简单,K8s 太复杂?w7panel 让容器管理更轻松!
点击右上角即可分享
微信分享提示