Python-119：优惠卷优先级恢复

问题描述

在一次热闹的促销活动中，小C策划了一项优惠券发放计划，准备了 nn 种不同的优惠券，编号为 11 到 nn。为了确保活动顺利进行，小C为优惠券制定了几条规则：

• 每位顾客最多只能领取其中一种优惠券。
• 每种优惠券都有一个独特的优先级，优先级之间没有重复。
• 只有满足某些特定条件的顾客才能领取对应的优惠券。
• 如果一位顾客符合多个优惠券的发放条件，他只会领取其中优先级最高的优惠券。

活动结束后，小C发现用于分析的优惠券优先级记录丢失了。他只能依靠活动期间留下的 mm 条顾客发放记录进行恢复。每条记录包含了顾客满足发放条件的所有优惠券类型及他最终获得的优惠券。现在小C希望你能帮忙恢复这些优惠券的优先级顺序。

如果存在多个可能的解，输出字典序最小的解。

需要注意的是，array表中，每个列表的第一个元素表示后面的抽奖券数，第二个元素表示顾客实际拿到的抽奖券。

代码

def solution(n, m, array):

    assert m == len(array)

    tmp = []

    for u in array:

        tmp += u

    assert max(tmp, default=0) <= n

    from heapq import heappush, heappop

    q = []

    g = [[] for _ in range(n + 1)]

    deg = [0 for _ in range(n + 1)]

    for u in array:

        u = u[1:]

        for i in range(1, len(u)):

            g[u[0]].append(u[i])

            deg[u[i]] += 1

    for i in range(1, n + 1):

        if deg[i] == 0:

            heappush(q, i)

    ans = []

    while q:

        u = heappop(q)

        ans.append(u)

        for v in g[u]:

            deg[v] -= 1

            if deg[v] == 0:

                heappush(q, v)

    assert len(ans) == n

    return ans

if __name__ == "__main__":

    # Add your test cases here

    print(solution(5, 3, [[3, 5, 4, 2], [3, 1, 3, 5], [2, 2, 4]]) == [1, 3, 5, 2, 4])

    print(solution(4, 2, [[2, 3, 4], [3, 2, 1, 3]]) == [2, 1, 3, 4])

    print(solution(3, 3, [[2, 2, 1], [2, 3, 1], [2, 3, 2]]) == [3, 2, 1])