剑指Offer：面试题09——用两个栈实现队列（java）

用两个栈实现队列

问题描述

• 用两个栈实现队列的数据结构（先进先出）

算法思想

• 要利用两个先入后出的栈去实现一个先入先出的队列，首先我们要明白的就是入队列和出队列肯定是使用两个不同的栈。说明两个栈一个是实现入队列的，一个是实现出队列的。

• 那我们就一步步的来观察：假设定义 input 栈和 output 栈两个栈

  • 1. 入队abc三个元素：就放入input栈中， 此时intput中有【a，b，c】（这时候不要管output栈，因为还没有到输出的时候，往往这时候去思考output就会打乱思绪）
  • 2. 出队a元素： 这时候我们要出input中栈底元素，很明显我们只需要将input中的元素顺序出栈压入output栈中即可！完成后，此时intput为空，output中有【c，b，a】，再进行output出栈即可。a元素出队列之后output中【c，b】。
  • 3. 入队d元素： 此时我们只需要将d元素压入input栈中即可，此时input【d】，output【c，b】
  • 4. 出队b元素： 此时因为output【c，b】，所以只需要出栈即可！

• 那我们现在的思路就很清楚了，每次入队只需要往input中加入元素，出队的时候要判断output是否为空，如果不为空直接出栈，如果为空要将input中的元素顺序压入output中再出栈！

• 下图为算法步骤图：
  

实现：

public class StackToQueue {

    private Stack<String> input;
    private Stack<String> output;


    public StackToQueue() {
        this.input = new Stack<String>();
        this.output = new Stack<String>();
    }


    //入队
    public void pushQueue(String x){

        this.input.push(x);

    }

    //出队
    public String popQueue(){

        if(this.output.isEmpty() && this.input.isEmpty()) 
            throw new IllegalArgumentException("队列长度为0");
        if(this.output.isEmpty()){ //判断output是否为空

            while(!this.input.isEmpty()){
                output.push(input.pop());
            }
        }


        return this.output.pop();

    }
}

总结

• 其实这道题不难，我也有自己的实现，但是比起这种方法来就显得不太行了
• 之所以没有想到这么优的解法是因为在入队的时候，脑子里就在构思出队的栈该如何存储，这时候反而会想多。所以我的算法就是每次入队都要将元素加入到output的栈底，那这样就大大复杂了算法！
• 所以以后思考步骤问题就慢慢来，想的太多反而会走弯路