package ch1;

import java.util.Stack;

/**
 * @Question: 用两个栈来假装一个队列
 * @Author: linjiayu
 * @Date: 2020/6/4 22:56
 */
public class TwoStackPretendQueue {

  //  一个栈用于接受元素进来
  Stack<Integer> stackOne;
  //  一个栈负责把元素抛出去
  Stack<Integer> stackTwo;

  public TwoStackPretendQueue() {
    this.stackOne = new Stack<>();
    this.stackTwo = new Stack<>();
  }

  //  Queue队列的特性便是尾进头出
  //  把元素摁到尾部
  public synchronized Integer addLast(Integer a) {
    stackOne.push(a);
    return a;
  }

  public synchronized Integer getFirst() throws Exception {
    //  取元素的栈为空，需要从放元素的栈拿元素出来
    if (stackTwo.isEmpty()) {
      while (!stackOne.isEmpty()) {
        stackTwo.push(stackOne.pop());
      }
    }
    //  如果还是没有元素
    if (stackTwo.isEmpty()) {
      throw new Exception("没有可以取的元素");
    }
    return stackTwo.pop();
  }

  public static void main(String[] args) throws Exception {
    TwoStackPretendQueue queue = new TwoStackPretendQueue();
    queue.addLast(1);
    queue.addLast(2);
    queue.addLast(3);
    System.out.println(queue.getFirst());
    System.out.println(queue.getFirst());
    queue.addLast(5);
    queue.addLast(4);
    queue.addLast(9);
    System.out.println(queue.getFirst());
    System.out.println(queue.getFirst());
    System.out.println(queue.getFirst());
  }

}