实现打印机队列2023 Q1 | 100分

286 阅读1分钟

题目描述

有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 次
输入格式为:
6
IN 1 2
IN 3 4
IN 1 5
OUT 1
OUT 3
OUT 3
复制代码
输出格式为:
3
2
NULL

解题思路:

  1. IN P NUM-表示往打印机P中塞打印文件,其中文件的优先级别是NUM
  2. OUT P- 表示从打印机P中打印文件,(如果打印机有文件、按照优先级打印,如果打印机没有文件、返回NULL)
  3. 输出的是OUT中,打印机的行为,打印的是事件发生的第x次;
  4. 输入不一定按照先IN、后OUT,也有可能IN与OUT是相互交错;

利用优先队列,存储每台打印机的打印队列,每次打印从对应优先队列出队即可

public static void main(String[] args) {
    Scanner scanner = new Scanner(System.in);
    int n = scanner.nextInt();
    scanner.nextLine();
    int index = 1;
    ArrayList<String> printRes = new ArrayList<>();
    PriorityQueue<PrinterBean>[] queues = new PriorityQueue[5];
    for (int i = 0; i < 5; i++) {
        queues[i] = new PriorityQueue<>(new Comparator<PrinterBean>() {
            @Override
            public int compare(PrinterBean o1, PrinterBean o2) {
                return o2.priority - o1.priority;
            }
        });
    }
    for (int i = 0; i < n; i++) {
        String inputItem = scanner.nextLine();
        String[] option = inputItem.split(" ");
        int tempIndex = Integer.parseInt(option[1]);
        if (option[0].equals("IN")) {
            queues[tempIndex].add(new PrinterBean(index, Integer.parseInt(option[2])));
        } else {
            if (!queues[tempIndex].isEmpty()) {
                PrinterBean bean = queues[tempIndex].poll();
                printRes.add(String.valueOf(bean.index));
            } else {
                printRes.add("NULL");
            }
        }
        index++;
    }

    for (int i = 0; i < printRes.size(); i++) {
        System.out.println(printRes.get(i));
    }

}

public static class PrinterBean {
    private int index;
    private int priority;

    public PrinterBean(int index, int priority) {
        this.index = index;
        this.priority = priority;
    }

    public int getIndex() {
        return index;
    }

    public void setIndex(int index) {
        this.index = index;
    }

    public int getPriority() {
        return priority;
    }

    public void setPriority(int priority) {
        this.priority = priority;
    }
}

用例通过率100%