2021天梯赛 L2-1 包装机（JAVA）

一种自动包装机的结构如图 1 所示。首先机器中有 N 条轨道，放置了一些物品。轨道下面有一个筐。当某条轨道的按钮被按下时，活塞向左推动，将轨道尽头的一件物品推落筐中。当 0 号按钮被按下时，机械手将抓取筐顶部的一件物品，放到流水线上。图 2 显示了顺序按下按钮 3、2、3、0、1、2、0 后包装机的状态。

图1 自动包装机的结构

图 2 顺序按下按钮 3、2、3、0、1、2、0 后包装机的状态

一种特殊情况是，因为筐的容量是有限的，当筐已经满了，但仍然有某条轨道的按钮被按下时，系统应强制启动 0 号键，先从筐里抓出一件物品，再将对应轨道的物品推落。此外，如果轨道已经空了，再按对应的按钮不会发生任何事；同样的，如果筐是空的，按 0 号按钮也不会发生任何事。

现给定一系列按钮操作，请你依次列出流水线上的物品。

输入格式：

输入第一行给出 3 个正整数 N（≤100）、M（≤1000）和 S​max​​（≤100），分别为轨道的条数（于是轨道从 1 到 N 编号）、每条轨道初始放置的物品数量、以及筐的最大容量。随后 N 行，每行给出 M 个英文大写字母，表示每条轨道的初始物品摆放。

最后一行给出一系列数字，顺序对应被按下的按钮编号，直到 −1 标志输入结束，这个数字不要处理。数字间以空格分隔。题目保证至少会取出一件物品放在流水线上。

输出格式：

在一行中顺序输出流水线上的物品，不得有任何空格。

输入样例：

3 4 4
GPLT
PATA
OMSA
3 2 3 0 1 2 0 2 2 0 -1

输出样例：

MATA

JAVA写好像只有20分，5分的运行时间超时了，最后拿C++写了一边，那个发布的人比较多，就不再发布了。总体上还是用栈。

附上代码

public static void main(String[] args){
        Scanner scanner = new Scanner(System.in);
        ArrayList<ArrayList<String>> list = new ArrayList<>();
        int n = scanner.nextInt();
        int m = scanner.nextInt();
        int s = scanner.nextInt();
        int flag = 0;
        scanner.nextLine();
        String[][] a = new String[n][m];
        int[] index = new int[n];
        ArrayList<String> l = new ArrayList<>();
        Stack<String> stack = new Stack<>();
        for (int i = 0; i < n; i++){
                String f = scanner.nextLine();
                a[i] = f.split("");
        }
        while (true){
            int d = scanner.nextInt();
            if (d == -1){
                break;
            }
            if (d != 0) {
                if (index[d-1] == m){
                    continue;
                }
                else if (stack.size() == s){
                    l.add(stack.peek());
                    stack.remove(stack.peek());
                    stack.add(a[d - 1][index[d - 1]]);
                    index[d - 1]++;
                }else {
                    stack.add(a[d - 1][index[d - 1]]);
                    index[d - 1]++;
                }
            }
            if (d == 0){
                if (stack.size() == 0 ){
                    continue;
                }else {
                    l.add(stack.peek());
                    stack.remove(stack.peek());
                }
            }
        }
        for (String value : l) {
            System.out.print(value);
        }
    }
}