一、描述
用两个栈实现一个队列,队列的声明如下,请实现它的两个函数appendTail和deleteHead,分别在队列尾部插入结点和队列头部删除结点的功能。
class CQueue<T> { private Stack<T> stack1; private Stack<T> stack2; public void appendTail(T node) { } public T deleteHead() { } }
二、解题思路
队列的规则是先进先出,而栈的规则是后进先出,由于有两个栈,可以让元素进栈两次,就可以模拟出先进先出的效果。可以让一个队列存放元素,另一个队列作为中转队列。
三、代码
package com.hust.grid.leesf.chapter2; import java.util.Stack; /** * Created by LEESF on 2016/8/29. */ class CQueue<T> { private Stack<T> stack1; private Stack<T> stack2; public CQueue() { stack1 = new Stack<T>(); stack2 = new Stack<T>(); } public void appendTail(T node) { stack1.push(node); } public T deleteHead() { if (!stack2.isEmpty()) { return stack2.pop(); } while (!stack1.isEmpty()) { stack2.push(stack1.pop()); } if (!stack2.isEmpty()) return stack2.pop(); return null; } } public class QueueWithTwoStacks { public static void main(String[] args) { CQueue<Integer> cQueue = new CQueue<Integer>(); cQueue.appendTail(1); cQueue.appendTail(2); cQueue.appendTail(3); System.out.println(cQueue.deleteHead()); cQueue.appendTail(4); System.out.println(cQueue.deleteHead()); System.out.println(cQueue.deleteHead()); System.out.println(cQueue.deleteHead()); } }
运行结果:
1 2 3 4