题解 | #火车进站#

火车进站

https://www.nowcoder.com/practice/97ba57c35e9f4749826dc3befaeae109

import java.util.ArrayList;
import java.util.Collections;
import java.util.Scanner;
import java.util.Stack;

public class Main {
    private static ArrayList<String> result = new ArrayList<>();

    public static void main(String[] args){
        Scanner in = new Scanner(System.in);

        while(in.hasNext()){
            solution(in);
        }
    }

    /**
     * 栈
     * @param in
     */
    private static void solution(Scanner in){
        int num = in.nextInt();
        int[] trains = new int[num];

        // 列车编号
        for(int i=0; i<num; i++){
            trains[i] = in.nextInt();
        }

        result = new ArrayList<>();

        // 火车站
        Stack<Integer> station = new Stack<>();

        // 发车
        depart(trains, station, 0, 0, "");

        Collections.sort(result);

        for(String seq: result){
            System.out.println(seq);
        }
    }

    /**
     * 递归
     * @param trains    列车初始编号序列
     * @param station   火车站
     * @param inNum     驶入火车站列车数
     * @param outNum    驶出火车站列车数
     * @param seqStr    驶出火车站列车序列
     */
    private static void depart(int[] trains, Stack<Integer> station, int inNum, int outNum, String seqStr){
        // 列车全部驶出
        if(outNum == trains.length){
            result.add(seqStr);
            return;
        }

        if(!station.isEmpty()){
            // 驶出
            Integer seq = station.pop();
            depart(trains, station, inNum, outNum+1, seqStr+seq+" ");
            // 恢复
            station.push(seq);
        }

        if(inNum < trains.length){
            // 驶入
            station.push(trains[inNum]);
            depart(trains, station, inNum+1, outNum, seqStr);
            // 恢复
            station.pop();
        }
    }
}

全部评论

相关推荐

07-09 19:25
门头沟学院 Java
这是要把每一个投校招的都开盒吗?
码农索隆:隐私这一块,直接是毫无隐藏😅
点赞 评论 收藏
分享
06-08 22:25
门头沟学院 Java
从零开始的转码生活:这hr不会打开手机不分青红皂白给所有人群发这句话,过一会再给所有人再发一遍,这肯定会有重复的,不管,再过一会再发一遍
点赞 评论 收藏
分享
07-07 12:47
门头沟学院 Java
码农索隆:竟然还真有卡体检报告的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务