华为OD机试 - 火车进站（Java 2024 E卷 100分）

原创

JavaPub 2024-11-26 10:25:44 博主文章分类：JavaPub ©著作权

文章标签 .net html java 面试 文章分类 Python 后端开发 yyds干货盘点

©著作权归作者所有：来自51CTO博客作者JavaPub的原创作品，请联系作者获取转载授权，否则将追究法律责任

华为OD机试 2024E卷题库疯狂收录中，刷题 <a rel="nofollow" href="https://blog.csdn.net/qq_40374604/category_12822407.html" style="color: red;">点这里</a>。

实战项目访问：http://javapub.net.cn/

专栏导读

本专栏收录于《华为OD机试（JAVA）真题（E卷+D卷+A卷+B卷+C卷）》。

刷的越多，抽中的概率越大，私信javapub，备注华为OD，加入华为OD刷题交流群，每一题都有详细的答题思路、详细的代码注释、3个测试用例、为什么这道题采用XX算法、XX算法的适用场景，发现新题目，随时更新，全天CSDN在线答疑。

火车进站

给定一个正整数N代表火车数量，接下来输入火车入站的序列，一共N辆火车，每辆火车以数字1-9编号，火车站只有一个方向进出，同时停靠在火车站的列车中，只有后进站的出站了，先进站的才能出站。要求输出所有火车出站的方案，以字典序排序输出。

进阶：时间复杂度： O(n!) ，空间复杂度： O(n)

输入描述：

第一行输入一个正整数N（0 < N < 10），第二行包括N个正整数，范围为1到10。

输出描述：

输出以字典序从小到大排序的火车出站序列号，每个编号以空格隔开，每个输出序列换行，具体见sample。

示例1

输入

3
1 2 3

输出

说明

第一种方案：1进、1出、2进、2出、3进、3出
第二种方案：1进、1出、2进、3进、3出、2出
第三种方案：1进、2进、2出、1出、3进、3出
第四种方案：1进、2进、2出、3进、3出、1出
第五种方案：1进、2进、3进、3出、2出、1出
请注意，[3,1,2]这个序列是不可能实现的。

Java 编程

package cn.net.javapub.demo2.demo;

/**
 * @author: shiyuwang
 * @url: http://javapub.net.cn
 */

import java.util.*;

public class Main {
    static Stack<String> stack = new Stack<String>();
    static Stack<String> list = new Stack<String>();
    static List<String> result = new ArrayList<>();

    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        while (in.hasNext()) {
            int n = in.nextInt();
            in.nextLine();

            String line = in.nextLine();
            process(line);
        }
    }

    public static void process(String s) {
        String[] strs = s.split("\\s");
        for (int i = strs.length - 1; i >= 0; i--) {
            list.push(strs[i]);
        }
        method("");
        Collections.sort(result);
        for (String r : result) {
            System.out.println(r);
        }
    }

    public static void method(String s) {
        if (stack.empty() && list.empty()) {
            result.add(s.trim());
            return;
        }
        if (!stack.empty()) {
            String str = stack.pop();
            method(s + " " + str);
            stack.push(str);
        }
        if (!list.empty()) {
            String str = list.pop();
            stack.push(str);
            method(s);
            stack.pop();
            list.push(str);
        }
    }
}

展示效果：

在这里插入图片描述