用两个栈实现队列

用两个栈实现队列

http://www.nowcoder.com/questionTerminal/54275ddae22f475981afa2244dd448c6

描述


这是一道对栈和队列之间灵活转化的题目。
难度:一星
考察知识:队列,栈


题解

方法:模拟

如果我知道队列是FIFO,栈是FILO,但是这道题我还是不知道怎么写怎么办?
对于这种感觉不难,但是又不会写的,方法就是模拟。
比如有如下操作:(pop操作确保栈中有元素)

push(1);push(2);pop(3);push(4);

图片说明
根据队列的特性,只能pop(1),pop(2),pop之后的结果
图片说明

上述是队列的操作。
当push的时候,我们必须要用一个stack来存,假设用stack1来存。
图片说明
那么push操作解决了。那么pop操作怎么办呢?
如果pop(1),但是此时在stack1的栈底,如果要pop,必须再将stack1中的数据push到stack2中,然后在pop,如图

这样直接弹出stack2的栈顶就可以了。
如果要继续pop,那就继续弹出stack2就可以了


但是现在总感觉哪里还是有点问题。如果是这样就继续测试几个例子。
如果push(5),
图片说明

所以最后总结一下:push操作就直接往stack1中push, pop操作需要分类一下:如果stack2为空,那么需要将stack1中的数据转移到stack2中,然后在对stack2进行pop,如果stack2不为空,直接pop就ok。

复杂度分析

时间复杂度:push操作为O(1),pop操作为O(1)
空间复杂度:需要stack来存,O(n)
##代码

class Solution
{
public:
    void push(int node) {
        stack1.push(node);
    }

    int pop() {
        if (stack2.empty()) {
            while (!stack1.empty()) {
                stack2.push(stack1.top());
                stack1.pop();
            }
        }
        int ret = stack2.top();
        stack2.pop();
        return ret;
    }

private:
    stack<int> stack1;
    stack<int> stack2;
};
全部评论
这tm叫O(1)???
19
送花
回复 分享
发布于 2021-10-26 20:04
关于 pop() 时间复杂度,应该是:最快为 O(1),最差为 O(n),n代表的是stack1 里面的元素个数
8
送花
回复 分享
发布于 2020-08-20 17:14
秋招专场
校招火热招聘中
官网直投
你这个先push后pop没问题,一会push,一会pop,一会又push,一会又pop。push和pop交替调用的时候就乱套了
8
送花
回复 分享
发布于 2021-07-28 23:53
这个用例也太模糊了吧
5
送花
回复 分享
发布于 2020-08-05 15:18
stack2.push(stack1.top()); stack1.pop(); 两句为什么不合为一句呢? stack2.push(stack1.pop());
1
送花
回复 分享
发布于 2021-08-20 11:55
时间复杂度对吗?
1
送花
回复 分享
发布于 2022-01-08 16:07
这个只能说是平均时间复杂度能达到O(1)了
1
送花
回复 分享
发布于 2022-01-08 22:42
每个元素最多入stack1一次,出stack1一次,入stack2一次,出stack2一次,所以O(1)。表面是循环,实际要根据元素来分析
1
送花
回复 分享
发布于 2022-02-25 11:27
为什么要return ret;
点赞
送花
回复 分享
发布于 2020-07-16 22:11
如果stack1本身为空呢 这种情况也要考虑吧
点赞
送花
回复 分享
发布于 2020-08-17 10:45
pop结束之后需要恢复stack1,否则pop后push再pop就有问题了
点赞
送花
回复 分享
发布于 2021-10-22 12:17
这个题有点误导人啊,这不是 O(1) 复杂度吧
点赞
送花
回复 分享
发布于 2022-01-06 16:47
出题出不清楚 再乱来只能搞乐扣去了
点赞
送花
回复 分享
发布于 2022-01-07 12:22

相关推荐

88 12 评论
分享
牛客网
牛客企业服务