PTA - 插松枝（模拟题）

PTA - 插松枝（模拟题）

人造松枝加工场的工人需要将各种尺寸的塑料松针插到松枝干上，做成大大小小的松枝。他们的工作流程（并不）是这样的：

• 每人手边有一只小盒子，初始状态为空。
• 每人面前有用不完的松枝干和一个推送器，每次推送一片随机型号的松针片。
• 工人首先捡起一根空的松枝干，从小盒子里摸出最上面的一片松针 —— 如果小盒子是空的，就从推送器上取一片松针。将这片松针插到枝干的最下面。
• 工人在插后面的松针时，需要保证，每一步插到一根非空松枝干上的松针片，不能比前一步插上的松针片大。如果小盒子中最上面的松针满足要求，就取之插好；否则去推送器上取一片。如果推送器上拿到的仍然不满足要求，就把拿到的这片堆放到小盒子里，继续去推送器上取下一片。注意这里假设小盒子里的松针片是按放入的顺序堆叠起来的，工人每次只能取出最上面（即最后放入）的一片。
• 当下列三种情况之一发生时，工人会结束手里的松枝制作，开始做下一个：

（1）小盒子已经满了，但推送器上取到的松针仍然不满足要求。此时将手中的松枝放到成品篮里，推送器上取到的松针压回推送器，开始下一根松枝的制作。

（2）小盒子中最上面的松针不满足要求，但推送器上已经没有松针了。此时将手中的松枝放到成品篮里，开始下一根松枝的制作。

（3）手中的松枝干上已经插满了松针，将之放到成品篮里，开始下一根松枝的制作。

现在给定推送器上顺序传过来的 N 片松针的大小，以及小盒子和松枝的容量，请你编写程序自动列出每根成品松枝的信息。

输入格式：

输入在第一行中给出 3 个正整数：N（≤103），为推送器上松针片的数量；M（≤20）为小盒子能存放的松针片的最大数量；K（≤5）为一根松枝干上能插的松针片的最大数量。

随后一行给出 N 个不超过 100 的正整数，为推送器上顺序推出的松针片的大小。

输出格式：

每支松枝成品的信息占一行，顺序给出自底向上每片松针的大小。数字间以 1 个空格分隔，行首尾不得有多余空格。

输入样例：

8 3 4
20 25 15 18 20 18 8 5

输出样例：

20 15
20 18 18 8
25 5

题解

模拟一类题，需要理解题目中的每一步，将题目的流程写下来好好分析一下。

1. 开始 盒子、松枝均为空（我们可以使用List<List<Integer>>来表示松枝的成品，盒子就可以用Stack<Integer>来模拟）
2. 使用数组模拟推送器
3. 先从盒子里拿松枝片，盒子没有从推送器上拿；且必须满足每颗松枝列表是单调递减的
4. 松枝成品时机
5. 松枝已满且推送器仍不满足需求（推送器上的松枝叶放在送至上不满足单调递减）
6. 盒子中的松枝片不满足需求且推送器上的松枝片不满足需求
7. 松枝上的松枝片以插满
8. 推送器上的松枝发放完毕，盒子总的是否已经发放完毕
9. 发放完毕：将最后一个松枝加入到成品送至列表里即可
10. 未发放完毕：遍历盒子总的松枝片，必须满足以上条件，将盒子中的松枝片使用完即可

代码

import java.util.*;

/**
 * @author sean
 * @version 1.0
 * @date 2024-3-13 20:19
 */
public class W3 {
    public static void main(String[] args) {
        Scanner scanner = new Scanner(System.in);
        // n 流水线上松枝片数量
        int n = scanner.nextInt();
        // m 框内最大片数
        int m = scanner.nextInt();
        // k 松枝最大片数
        int k = scanner.nextInt();

      	// 松枝
        List<Integer> pb = new ArrayList<>();	
      	// 成品松枝列表
        List<List<Integer>> res = new ArrayList<>();
      	// 盒子
        Stack<Integer> box = new Stack<>();
      	// 推送器
        int[] arr = new int[n];
        for (int i = 0; i < arr.length; i++) {
            arr[i] = scanner.nextInt();
        }
      
        int index = 0;
        // 处理控制台和盒子中的松枝片
        while (index < arr.length) {
            // 盒子不是空的 从盒子中取
            if (!box.isEmpty()) {
                if (pb.isEmpty()) {
                    pb.add(box.pop());
                }
                // 从盒子中取出合适的松针片
                while (!box.isEmpty() && box.peek() <= pb.get(pb.size() - 1)) {
                    // 取的过程中需要检查松针上的松针片是否已满
                    if (pb.size() == k) {
                        break;
                    }
                    pb.add(box.pop());
                }
            }
            // 松枝上的松枝片已满
            if (pb.size() == k) {
                res.add(pb);
                pb = new ArrayList<>();
                continue;
            }
            // 盒子满了且操作台上仍不满足将松枝片插入松枝
            if (box.size() == m && arr[index] > pb.get(pb.size() - 1)) {
                res.add(pb);
                pb = new ArrayList<>();
                continue;
            }
            // 松枝为空或者松枝为空 直接拿控制台上的
            if (box.isEmpty() && pb.isEmpty()) {
                pb.add(arr[index++]);
                continue;
            }
            // 操作台松枝片 大于 松枝最上的松枝片
            if (arr[index] > pb.get(pb.size() - 1)) {
                // 放入盒子
                box.push(arr[index]);
            } else {
                // 插到松枝上
                pb.add(arr[index]);
            }
            index++;
        }

        // 控制台没有松枝片 将其放入成品篮中
        if (pb.size() == k) {
            res.add(pb);
            pb = new ArrayList<>();
        }

        // 处理盒子中的松枝片
        if (!pb.isEmpty()) {
            if (!box.isEmpty()) {
                if (pb.get(pb.size() - 1) < box.peek()) {
                    res.add(pb);
                    pb = new ArrayList<>();
                } else {
                    pb.add(box.pop());
                }
            }
        }
      	// 遍历盒子中的松枝片
        while (!box.isEmpty()) {
          	// 还是上面的套路，只不过不需要再判断盒子中是否超出了
            if (pb.isEmpty()) {
                pb.add(box.pop());
                continue;
            }
            if (pb.size() == k || box.peek() > pb.get(pb.size() - 1)) {
                res.add(pb);
                pb = new ArrayList<>();
                continue;
            }
            pb.add(box.pop());
        }
      	// 将最后一个松枝放在成品列表中
        if (!pb.isEmpty()) {
            res.add(pb);
        }
      	// 遍历松枝列表即可
        for (List<Integer> re : res) {
            for (int i = 0; i < re.size() - 1; i++) {
                System.out.print(re.get(i) + " ");
            }
            System.out.println(re.get(re.size() - 1));
        }
    }
}