LeetCode---Implement Queue using Stacks

题目大意:用栈实现队列,包括入队列,出队列,取队头,判空等方法。

算法思想:

1.用两个栈模拟队列,一个栈s1模拟队列的尾部,一个栈s2模拟队列的头部。

2.当入队列是,只需将元素压入s1即可。若s1和s2都为空则队列为空。

3.当出队列时,先判断队列是否为空,若不空则看s2是否为空,不空则去栈顶元素即可,若空则将s1中的元素都压入s2栈中,然后取s2的栈顶元素。

4.同理3可实现队列的出队列操作。

代码如下:

class Queue {
public:
    stack<int> s1,s2;
    // Push element x to the back of queue.
    void push(int x) {
        s1.push(x);
    }

    // Removes the element from in front of queue.
    void pop(void) {
        if(empty()) return ;
        if(!s2.empty())
             s2.pop();
        else{
            while(!s1.empty()){
                s2.push(s1.top());
                s1.pop();
            }
            s2.pop();
        }
    }

    // Get the front element.
    int peek(void) {
        if(empty()) return 0;
        if(!s2.empty())
             return s2.top();
        else{
            while(!s1.empty()){
                s2.push(s1.top());
                s1.pop();
            }
            return s2.top();
        }
    }

    // Return whether the queue is empty.
    bool empty(void) {
        return s1.empty()&&s2.empty();
    }
};


评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值