题目
编写一个类,用两个栈实现队列,支持队列的基本操作:add,poll,peek.
思路
栈的特点是先进后出,那么设有两个栈,第一个栈命名为stackPush,第二个栈命名为stackPop。假设有1,2,3,4,5依次进栈stackPush,然后1-5全部出栈到stackPop,此时stackPop的栈顶元素就是1了,此时出栈的第一个元素就是当前栈顶1了,也就是符合队列的先进先出特性了。
乍一看好像很简单的样子其实不然
第一种情况StackPush的元素未完全倒入StackPop,会导致本该在队列中部的元素先出队列
第二种情况StackPop不未空时就向stackPop内导入元素则打乱了整个队列的连续性,使得本该在队尾的一段元素与本该在队前的元素颠倒。
如何规避呢?请看代码实现
public class queuebystack {
private Stack<Integer> stackPush;
private Stack<Integer> stackPop;
public queuebystack() {
this.stackPush = new Stack<Integer>();
this.stackPop = new Stack<Integer>();
}
public void pushToPop() {
if (stackPop.empty()) {
// 只有为空时才倒入
while (!stackPush.empty()) {
stackPop.push(stackPush.pop());
}
}
}
public void add(int queueIn) {
stackPush.push(queueIn);
pushToPop();// 只有stackPop为空时,才会向其导入,保存队列连贯性
}
public int poll() {
if (stackPush.empty() && stackPop.empty()) {
throw new RuntimeErrorException(null, "队列为空!");
}
pushToPop();// 如果还没有像stackPop中倒入;
return stackPop.pop();
}
public int peek() {
if (stackPush.empty() && stackPop.empty()) {
throw new RuntimeErrorException(null, "队列为空!");
}
pushToPop();// 如果还没有像stackPop中倒入;
return stackPop.peek();
}
}
图解模拟
在poll和过程中都会调用pushToPop这个函数,这个函数会判断stackPop是否为空,为空是才表明现阶段符合队列顺序的元素都已出队,不为空时,用stackPush保存下来,直到stackPop为空才一次性导入stackPop。
总结
我想这道题应该主要培养的还是程序员缜密的思维吧,要尽可能考虑到所有可能出现的情况。