Acwing129.火车进栈(Java实现)——使用dfs+栈+队列

题目概述：
这里有 n 列火车将要进站再出站，但是，每列火车只有 1 节，那就是车头。

这 n 列火车按 1 到 n 的顺序从东方左转进站，这个车站是南北方向的，它虽然无限长，只可惜是一个死胡同，而且站台只有一条股道，火车只能倒着从西方出去，而且每列火车必须进站，先进后出。

也就是说这个火车站其实就相当于一个栈，每次可以让右侧头火车进栈，或者让栈顶火车出站。

车站示意如图：

            出站<——    <——进站
                     |车|
                     |站|
                     |__|

现在请你按《字典序》输出前 20 种可能的出栈方案。

算法思路：
分成 3 个状态之间的转换 , 第一个状态就是火车出站序列（queue）, 第二个状态就是火车在站内的序列（stack） ,第三个状态就是火车在还没入站时的状态。
那么我们要求出站的可能序列,而且还是按照字典序 , 怎么保证呢? 火车在车站时有两种选择, 第一个就是车站的栈顶火车出站 ,第二个选择就是 火车从入站状态到 在车站状态, 为了保证是字典序,我们应该先dfs第一种选择, 然后dfs第二种选择, 为什么? 因为我们火车是按照编号递增进入火车站的, 所以前面第一种选择的火车出站后的编号一定比 火车入站的编号大 . 因为是按照顺序进站的,所以用一个int 变量表示就行 .

ps:
本次算法实现的难点是在，回溯队列当中队尾元素的时候，在Java当中是无法直接取出队尾元素的，那么我们需要依次先把队头元素取出再入队，直到取到size()-1索引处的元素时，不入队即可。

回溯队尾元素代码实现：

            //进行回溯(回溯的时候可以将队列（尾部）当中的元素进行删除)
            int len = queue.size();
            int a = 0;
            for(int i=0;i<len;i++){
                int head = queue.poll();
                if(i!=len-1){
                    queue.add(head);
                }else{
                    //将队列对尾元素取出后赋值给a
                    a = head;
                }
            }
            stack.push(a);

本题算法实现：

import java.util.*;
class Main{
    //定义一个队列用来存储出站的顺序
    public static Queue<Integer> queue = new LinkedList<>();
    //定义一个栈
    public static Stack<Integer> stack = new Stack<>();
    public static int count = 20;
    public static int num;
    //表示当前遍历到哪一列火车了
    public static int n = 1;
    public static void main(String[] args){
        Scanner scanner = new Scanner(System.in);
        //获得火车的数量
        num = scanner.nextInt();
        dfs();
        
    }
    
    //使用深度优先遍历
    public static void dfs(){
        //结束递归
        if(count==0){
            return;
        }
        //当队列满的时候可以将答案输出
        if(queue.size()==num){
            count--;
            String res = "";
            //将队列当中的元素取出
            for(int i=0;i<num;i++){
                int k = queue.poll();
                res+=String.valueOf(k);
                queue.add(k);
            }
            System.out.println(res);
            return;
        }
        //如果栈当中不为空
        if(!stack.isEmpty()){
            //将栈顶元素加入到队列当中
            queue.add(stack.pop());
            //进行深度优先遍历
            dfs();
            //进行回溯(回溯的时候可以将队列（尾部）当中的元素进行删除)
            int len = queue.size();
            int a = 0;
            for(int i=0;i<len;i++){
                int head = queue.poll();
                if(i!=len-1){
                    queue.add(head);
                }else{
                    //将队列对尾元素取出后赋值给a
                    a = head;
                }
            }
             stack.push(a);
        }
        //表示当前还有火车需要进栈
        if(n<=num){
            //将当前车进栈
            stack.push(n);
            n++;
            //进行深度优先遍历
            dfs();
            //回溯
            n--;
            stack.pop();
        }
    }
}