打印文件/打印机队列 - 华为OD机试真题(Java 题解)-优快云博客

本文链接：https://blog.youkuaiyun.com/user_longling/article/details/149334620

华为OD机试题库《C++》限时优惠 9.9

华为OD机试题库《Python》限时优惠 9.9

华为OD机试题库《JavaScript》限时优惠 9.9

针对刷题难，效率慢，我们提供一对一算法辅导，针对个人情况定制化的提高计划（全称1V1效率更高）。

看不懂有疑问需要答疑辅导欢迎私VX： code5bug

华为OD机试真题

题目描述

有5台打印机打印文件，每台打印机有自己的待打印队列。

因为打印的文件内容有轻重缓急之分，所以队列中的文件有1~10不同的代先级，其中数字越大优先级越高。

打印机会从自己的待打印队列中选择优先级最高的文件来打印。

如果存在两个优先级一样的文件，则选择最早进入队列的那个文件。

现在请你来模拟这5台打印机的打印过程。

输入描述

每个输入包含1个测试用例，

每个测试用例第一行给出发生事件的数量N（0 < N < 1000）。

接下来有 N 行，分别表示发生的事件。共有如下两种事件：

“IN P NUM”，表示有一个拥有优先级 NUM 的文件放到了打印机 P 的待打印队列中。（0< P <= 5, 0 < NUM <= 10)；
“OUT P”，表示打印机 P 进行了一次文件打印，同时该文件从待打印队列中取出。（0 < P <= 5）。

输出描述

对于每个测试用例，每次”OUT P”事件，请在一行中输出文件的编号。
如果此时没有文件可以打印，请输出”NULL“。
文件的编号定义为”IN P NUM”事件发生第 x 次，此处待打印文件的编号为x。编号从1开始。

示例1

输入：

7
IN 1 1
IN 1 2
IN 1 3
IN 2 1
OUT 1
OUT 2
OUT 2

输出：

3
4
NULL

示例2

输入：

5
IN 1 1
IN 1 3
IN 1 1
IN 1 3
OUT 1

输出：

题解

本题属于 模拟题，主要考察对优先级队列（堆）的应用。需要模拟多个打印机的打印队列，并根据优先级规则处理文件的入队和出队操作。

解题思路

数据结构选择：
使用 优先级队列（堆） 来维护每个打印机的待打印文件，确保能高效获取优先级最高的文件。
优先级规则：
优先级高的先打印（数值越大优先级越高）。
优先级相同则先入队的先打印（文件编号小的先打印）。

处理事件：
IN P NUM：将文件加入打印机 P 的队列，记录文件编号（全局自增）。
OUT P：从打印机 P 的队列中取出优先级最高的文件打印；若队列为空，输出 NULL。

Java

import java.util.*;

/**
 * @author code5bug
 */
public class Main {
    public static void main(String[] args) {
        Scanner in = new Scanner(System.in);
        // 内容项： int[] ｛优先级，文件编号｝
        List<PriorityQueue<int[]>> todoList = new ArrayList<>();

        // 初始化出 5 个打印队列
        for (int i = 0; i <= 5; i++) {
            todoList.add(new PriorityQueue<>((o1, o2) -> {
                if (o1[0] != o2[0]) { // 优先级降序
                    return o2[0] - o1[0];
                } else {    // 文件编号升序
                    return o1[1] - o2[1];
                }
            }));
        }

        // 文件编号
        int x = 0;
        // 测试用例数
        int N = Integer.parseInt(in.nextLine());
        for (int i = 0; i < N; i++) {
            String[] pairs = in.nextLine().split(" ");
            int p = Integer.parseInt(pairs[1]);
            // 放入打印队列
            if (pairs[0].equals("IN")) {
                int num = Integer.parseInt(pairs[2]);
                todoList.get(p).add(new int[]{num, ++x});
            } else {    // 打印文件
                PriorityQueue<int[]> que = todoList.get(p);
                if (que.isEmpty()) {
                    System.out.println("NULL");
                } else {
                    int[] poll = que.poll();
                    System.out.println(poll[1]);
                }
            }
        }
    }
}