算法总结之 两个栈组成的队列
编写一个类,用两个栈实现队列,支持队列的基本操作(add poll peek)
栈的特点是先进后出,而队列是先进先出。用两个栈正好能把顺序反转过来实现类似队列的操作
具体实现:
一个栈作为压入栈,在压入数据时候,在压入数据时只往这个栈中压入,记为 stackPush
另一个栈只作为弹出栈,在弹出数据时只从这个栈弹出,记为 stackPop
1 如果stackPush要往stackPop中压入数据, 那么必须一次性把stackPush中的数据全部压入
2 如果stackPop不为空,stackPush 绝对不能向stackPop中压入数据
那么 压入数据的操作在何时发生呢?
这个选择的时机有很多,调用 add poll peek 三种方法中的任何一种时发生 压入数据 都是可以的
看代码:
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 | package TT; import java.util.Stack; public class Test121 { public class TwoStacksQueue{ public Stack<Integer> stackPush; public Stack<Integer> stackPop; public TwoStacksQueue(){ stackPush= new Stack<Integer>(); stackPop = new Stack<Integer>(); } public void add( int pushInt){ stackPush.push(pushInt); //一个用来压入数据用的 } public int poll(){ if (stackPop.empty() && stackPop.empty()){ throw new RuntimeException( "queue is empty!" ); } else if (stackPop.empty()){ while (!stackPush.empty()){ stackPop.push(stackPush.pop()); } } return stackPop.pop(); } public int peek(){ if (stackPop.empty() && stackPush.empty()){ throw new RuntimeException( "queue is empty!" ); } else if (stackPop.empty()){ while (!stackPush.empty()){ stackPop.push(stackPop.pop()); } } return stackPop.peek(); } } } |
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步