用两个栈来实现一个队列-Java

用两个栈实现队列

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

一. 思路

栈是先进后出,队列是先进先出。不知道怎么解决,可以模拟一下。

目标是实现一个队列,如下:
图片说明

若用栈存储,如下:
图片说明

第二个图是核心解题思想。

二. 代码

import java.util.Stack;

public class Solution {
    Stack<Integer> stack1 = new Stack<Integer>();
    Stack<Integer> stack2 = new Stack<Integer>();

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

    public int pop() {
        if(stack2.size() <= 0){
            while(stack1.size() != 0){
                 stack2.push(stack1.pop());
            }
        }
        return stack2.pop();
    }
}

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

三. 复杂度分析

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

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务