利用两个栈实现队列的操作
`
题目:编写一个类,用两个栈实现队列,支持队列的基本操作(add,poll,peek)
解答:栈:先进后出
队列:先进先出
设置两个栈,分别为stackPush 和stackPop,先将数据压入stackPush栈,然后数据按照先进后出规则弹出,弹出的数据再顺序压入stackPop栈,再将stackPop栈的数据弹出,这样下来,总体可看做一个先进先出的队列,从而实现目标。
下边为具体的代码实现:
package com.thread.ThreadTest1.Singleton.TwoStacksQueue;
import java.util.Stack;
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()&&stackPush.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(stackPush.pop());
}
}
return stackPop.peek();//返回弹出栈的顶端
}
}