#P14836. [THUPC 2026 初赛] 能量分配

[THUPC 2026 初赛] 能量分配

题目背景

来自 2026 清华大学学生程序设计竞赛暨高校邀请赛(THUPC2026)初赛。

题解等资源可在 https://gitlink.org.cn/thusaa/thupc2026pre 查看。

题目描述

小 Q 是星际小学光速飞船课的老师,她准备向学校的星际能源商店购买 nn本质不同的能量石,其中每个能量石具有一单位能量,且每个能量石有一个 11nn 之间的正整数编号,每个能量石的编号互不相同。她想要把它们分给 mm 名学生,帮助学生们练习如何驾驶光速飞船。

在分配时,每个能量石会恰好被分给一名学生,且允许学生没有分配到任何能量石。定义 cic_i 表示第 ii 名学生分配到的能量石总数,即满足 i=1mci=n\sum_{i=1}^{m}c_i=nci0c_i\geq 0。两个能量石的分配方案本质不同当且仅当存在一个学生在两个方案中分配到的能量石的编号集合不同。例如对于两个能量石与两名学生,能量石 11 分给学生 11,能量石 22 分给学生 22 与能量石 11 分给学生 22,能量石 22 分给学生 11 被视为两种本质不同的方案,由于学生 11 在方案一中分配到的能量石的编号集合为 {1}\{1\} 而在方案二中分配到的能量石的编号集合为 {2}\{2\},所以被视为本质不同的方案。

然而,每名学生都想要自己分配到的用于光速飞船的能量石比其他学生分到的多,因此如果一个学生分配到的能量石比其他学生分到的能量石要更多,那么他会将拥有更大的满意度。

初始时每名学生的满意度为 11,则在分配完 nn 个能量石后第 ii 名学生的满意度可以按照如下方式计算:

  • 依次考虑每一个除了 ii 自己以外的学生 jj,若 ci>cjc_i>c_j,则 ii 的满意度将会变为原来的 Ai,jA_{i,j} 倍;若 ci=cjc_i=c_j,则 ii 的满意度将会变为原来的 Bi,jB_{i,j} 倍;若 ci<cjc_i<c_j,则 ii 的满意度将不会发生变化。保证满足 1<Bi,j<Ai,j1091<B_{i,j}<A_{i,j}\leq 10^9

对于第 ii 名学生,令其在分配完 nn 个能量石后的满意度为按照如上方式考虑完除自己以外的 m1m-1 名学生后最终的满意度。现在小 Q 定义一个能量石的分配方案的权值为在分完 nn 个能量石后 mm 名学生满意度之积,小 Q 想知道所有本质不同的能量分配方案的权值和317317 取模的结果。然而小 Q 是光速飞船课老师,不是光速计算课老师,所以她希望精通光速计算的您,迅速告诉她这个结果对 317317 取模的结果。

此外,由于小 Q 事先没有测试星际小学光速飞船的具体性能,当光速飞船的性能不同时,能量需求会发生变化。所以她会询问 qq 个可能购买的能量石总数 nn,对于每一个 nn,您需要告诉她所有若她购买了 nn 个能量石,所有能量石的分配方案的权值和对 317317 取模的结果。

输入格式

从标准输入读入数据。

第一行输入一个正整数 m(1m12)m(1\leq m\leq 12),表示学生总数。

接下来的 mm 行输入 m×mm\times m 的矩阵 AA,其中每行 mm 个正整数,第 ii 行第 jj 列表示 Ai,jA_{i,j} 的取值,其中当 i=ji=jAi,j=0A_{i,j}=0,当 iji\not=j 时有 1<Ai,j1091<A_{i,j}\leq 10^9

接下来的 mm 行输入 m×mm\times m 的矩阵 BB,其中每行 mm 个正整数,第 ii 行第 jj 列表示 Bi,jB_{i,j} 的取值,其中当 i=ji=jBi,j=0B_{i,j}=0,当 iji\not=j 时有 1<Bi,j<Ai,j1<B_{i,j}<A_{i,j}

之后一行输入一个正整数 q(1q100)q(1\leq q\leq 100),表示 qq 个询问。

接下来的 qq 行每行输入一个正整数 n(1n1018)n(1\leq n\leq 10^{18}),表示询问当能量石总数为 nn 时原问题的答案。

输出格式

输出到标准输出。

输出 qq 行,每行一个 [0,317)[0,317) 之间的整数,表示能量石的分配方案权值和对 317317 取模的结果。

3
0 7 5
5 0 7
4 6 0
0 4 4
3 0 4
3 2 0
4
1
2
10
1000000000000000000
37
303
46
148

提示

样例一解释

对于 n=1n=1,共有 33 种本质不同的能量石的分配方案,即将唯一的能量石分配给学生 1,21,233,每种方案的权值分别为 280,420,288280,420,288,所有分配方案的权值和为 988988,对 317317 取模的结果为 3737

对于 n=2n=2,共有 99 种本质不同的能量石的分配方案,对于每一组满足 1i,j31\leq i,j\leq 3 的正整数对 (i,j)(i,j),能量石 11 分配给学生 ii,能量石 22 分配给学生 jj 对应一组方案,各方案的权值如下表,其中第 ii 行第 jj 列的数表示将能量石 11 分配给学生 ii,能量石 22 分配给学生 jj 所对应的分配方案的权值:

280280 420420 504504
420420 160160
504504 160160 288288

各方案的权值和为 31563156,对 317317 取模的结果为 303303