题解 | #火车进站#

火车进站

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

import java.util.*;
import java.util.Scanner;
import java.util.function.Function;
import java.util.stream.Collectors;

public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        while (in.hasNextInt()) {
            int n = in.nextInt();
            int[] trains = new int[n];
            for (int i = 0; i < n; i++) {
                trains[i] = in.nextInt();
            }

            List<List<Integer>> result = new ArrayList<>();
            Deque<Integer> stack = new ArrayDeque<>();
            List<Integer> record = new ArrayList<>();

            backtrack(result, stack, trains, record, 0);

            // 排序并输出
            List<String> resultStrs = result.stream().map(integers -> {
                StringBuilder sb = new StringBuilder();
                integers.forEach((i) -> sb.append(i).append(' '));
                return sb.toString();
            }).sorted().collect(Collectors.toList());
            resultStrs.forEach(System.out::println);
        }
    }

    public static void backtrack(List<List<Integer>> result, Deque<Integer> stack, int[] trains,
                                 List<Integer> record, int index) {
        // 结束条件:所有车辆都入站过、车站内也没车辆、所有车辆都已经出站
        if (index == trains.length && stack.isEmpty() && record.size() == trains.length) {
            result.add(new ArrayList<>(record));
            return;
        }
        
        // 对于每一个状态,要么选择从车辆队列中入站,要么选择从车站中出站,
        // 如果选择入站,状态变化:车辆队列 index+1、站内车辆增加、出站的车辆不变
        // 如果选择出站,状态变化:车辆队列不变、站内车辆减少、出站的车辆增加

        // 出站,站内有车才能出站
        if (!stack.isEmpty()) {
            record.add(stack.pop());
            backtrack(result, stack, trains, record, index);
            stack.push(record.remove(record.size() - 1));
        }

        // 入站,车辆队列还有车才能入站
        if (index < trains.length) {
            stack.push(trains[index]);
            backtrack(result, stack, trains, record, index + 1);
            stack.pop();
        }
    }
}

全部评论

相关推荐

凉风落木楚山秋:1.教育背景,这一栏用于说明你是哪个省份的,一般四非在省内认可度是高于外省的,但是到外边了基本路边一条。然后这一栏可以写一写校内荣誉补充满一行 2.个人介绍没必要,把下面的技能内容放到这里面来,个人情况挖空等面试官来问你 3.竞赛奖项单独一栏,专门用2-3行小字写你获得了哪些奖 4.后面的模块就分为实习经历和项目经历,一个写实习做的项目,一个写学校做的项目。另外在项目中承担的角色可以标出来,时间周期也可以写一下 5.其他,你的经历和我挺像,我也大二的时候做前端,看你想找小程序还是web方向的。做小程序我感觉你这简历已经够了,比我的水货学弟强多了。要做web的话尽量再写一个react项目,不然走不远。另外,那个排课的项目看起来好水,没有亮点,可以再挖掘一下 6.名称,技术名称书写风格不统一,要么统一大驼峰,要么就用全小写+空格,保持一致好看很多
点赞 评论 收藏
分享
你背过凌晨4点的八股文么:简历挂了的话会是流程终止,像我一样
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务