【面试】用两个栈实现队列

一、描述

　　用两个栈实现一个队列，队列的声明如下，请实现它的两个函数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());
    }
}