【华为OD机试 2023】 打印机队列（Python）

题目描述

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

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

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

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

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

输入描述

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

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

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

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

输出描述

对于每个测试用例，每次”OUT P”事件，请在一行中输出文件的编号。

如果此时没有文件可以打印，请输出“NULL”

文件的编号定义为”IN P NUM”事件发生第 x 次，此处待打印文件的编号为x。编号从 1 开始。

用例

输入

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

输出

3
4
NULL

Python代码

n = int(input())
events = [input().split() for _ in range(n)]

printers = {}
for i in range(len(events)):
    event = events[i]
    if event[0] == 'IN':
        p = int(event[1])    # 哪台打印机
        num = int(event[2])  # 文件优先级
        if p not in printers:
            printers[p] = [(i + 1, num)]
        else:
            printers[p].append((i + 1, num))
    elif event[0] == 'OUT':
        p = int(event[1])    # 哪台打印机
        if p not in printers or len(printers[p]) == 0:
            print('NULL')
        else:
            printers[p].sort(key=lambda x:(-x[1], x[0]))
            # print(printers[p])
            print(printers[p].pop(0)[0])