用两个栈来实现一个队列-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)