Lintcode 用栈实现队列

用栈实现队列
正如标题所述,你需要使用两个栈来实现队列的一些操作。

队列应支持push(element),pop() 和 top(),其中pop是弹出队列中的第一个(最前面的)元素。

pop和top方法都应该返回第一个元素的值。

开始直观的想法:
`
class Queue {
public:
stack stack1;
stack stack2;

Queue() {
    // do intialization if necessary
}

void push(int element) {
    // write your code here
    stack1.push(element);
}

int pop() {
    // write your code here
    if(stack1.empty())
        return -1;
    while(!stack1.empty()){
        int num = stack1.top();
        stack1.pop();
        stack2.push(num);
    }
    //operate pop()
    int value = stack2.top();
    stack2.pop();
    while(!stack2.empty()){
        int num = stack2.top();
        stack2.pop();
        stack1.push(num);
    }
    return value;
}

int top() {
    // write your code here
    if(stack1.empty())
        return -1;
    while(!stack1.empty()){
        int num = stack1.top();
        stack1.pop();
        stack2.push(num);
    }
    //operate top()
    int value = stack2.top();
    while(!stack2.empty()){
        int num = stack2.top();
        stack2.pop();
        stack1.push(num);
    }
    return value;

}

};
`
挑战:仅使用两个栈来实现它,不使用任何其他数据结构,push,pop 和 top的复杂度都应该是均摊O(1)的

class Queue {
public:
stack stack1;
stack stack2;

Queue() {
    // do intialization if necessary
}

void push(int element) {
    // write your code here
    stack1.push(element);
}

int pop() {
    // write your code here
    if(stack1.empty()&&stack2.empty())
        return -1;
    if(stack2.empty()){
        while(!stack1.empty()){
            int num = stack1.top();
            stack1.pop();
            stack2.push(num);
        }

    }
    int value = stack2.top();
    stack2.pop();
    return value;
}

int top() {
    // write your code here
    if(stack1.empty()&&stack2.empty())
        return -1;
    if(stack2.empty()){
        while(!stack1.empty()){
            int num = stack1.top();
            stack1.pop();
            stack2.push(num);
        }

    }
    int value = stack2.top();
    return value;

}

};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值