#P11292. 【MX-S6-T4】「KDOI-11」彩灯晚会

    ID: 10750 Type: RemoteJudge 1000ms 512MiB Tried: 0 Accepted: 0 Difficulty: 6 Uploaded By: Tags>动态规划,dpO2优化动态规划优化拓扑排序组合数学容斥

【MX-S6-T4】「KDOI-11」彩灯晚会

题目背景

原题链接:https://oier.team/problems/96

题目描述

小 K 家的房子还蛮大的,所以他举办了彩灯晚会。

彩灯晚会显然有彩灯。于是小 K 找来了 nn 盏彩灯。

小 K 不希望彩灯散在地上,于是他用 mm有向边连接了这些彩灯,使得他们连在一起。也就是说,在将有向边看成无向边的情况下,这些彩灯是连通的。保证该图为有向无环图

小 K 的彩灯很厉害,每个都可以独立地发出 kk 种颜色中的任意一种。小 K 作为世界顶尖 Light Glowing Master(简称 LGM),决定将所有方案(共 knk^n 种)都尝试一下。

小 K 不喜欢很多的颜色,于是他给定了一个正整数 ll,并定义一个方案的美丽度为每种颜色长度为 ll 的链数量的平方之和。形式化地讲,设第 ii 种颜色的长度为 ll 的链的数量为 cnticnt_i,则一个方案的美丽度为 i=1kcnti2\sum_{i=1}^kcnt_i^2

现在小 K 想知道所有 knk^n 种方案的美丽度之和,对 998244353998244353 取模。

两种方案是不同的当且仅当存在某个彩灯在两种方案中发出不同的颜色。

一条长度为 ll 的链是一个 2l12l-1 元组 (v1,e1,v2,e2,,el1,vl)(v_1,e_1,v_2,e_2,\dots,e_{l-1},v_l),使得对于任意 1i<l1\leq i<l,第 eie_i 条有向边是由 viv_i 连向 vi+1v_{i+1} 的。两条长度为 ll 的链 (v1,e1,v2,e2,,el1,vl)(v_1,e_1,v_2,e_2,\dots,e_{l-1},v_l)(v1,e1,v2,e2,,el1,vl)(v'_1,e'_1,v'_2,e'_2,\dots,e'_{l-1},v'_l) 不同当且仅当存在 1i<l1\leq i<l 满足 eieie_i\neq e'_i 或存在 1il1\leq i\leq l 满足 viviv_i\neq v'_i

一种颜色 xx 的长度为 ll 的链是一条长度为 ll 的链 (v1,e1,v2,e2,,el1,vl)(v_1,e_1,v_2,e_2,\dots,e_{l-1},v_l) 使得对于任意 1il1\leq i\leq lviv_i 编号的彩灯发出颜色为 xx

输入格式

第一行,五个非负整数 id,n,k,l,Mid,n,k,l,M,其中 idid 表示测试点编号(所有样例满足 id=0id=0),nn 表示彩灯数量,kk 表示颜色数量,ll 表示链的长度,MM 的意义如后文所示。

接下来 MM 行,第 ii 行三个正整数 ui,vi,ciu_i,v_i,c_i,表示有 cic_iuiviu_i\to v_i 的有向边。也就是说,第 (j=1i1cj)+1\bigl(\sum_{j=1}^{i-1}c_j\bigr)+1 至第 (j=1icj)\bigl(\sum_{j=1}^ic_j\bigr) 条有向边由 uiu_i 连向 viv_i

题面中 mm 满足 m=i=1Mcim=\sum_{i=1}^Mc_i。保证:(u,v)(u,v) 互不相同,给定图为有向无环图,在将有向边看成无向边的情况下,彩灯是连通的。

输出格式

仅一行,一个非负整数,表示所有 knk^n 种方案的美丽度之和,对 998244353998244353 取模。

0 3 2 2 2
1 3 1
3 2 1
12
0 5 4 3 7
1 4 4
1 5 2
4 3 1
5 3 2
3 2 3
4 5 1
5 2 2
16096

提示

【样例解释 #1】

共有 23=82^3=8 种不同的彩灯显示颜色的方案,如下表所示(不妨假设颜色为黑和白):

编号 11 号彩灯颜色 22 号彩灯颜色 33 号彩灯颜色 美丽度
11 44
22 00
33 11
44
55
66
77 00
88 44

美丽度之和为 4+0+1+1+1+1+0+4=124+0+1+1+1+1+0+4=12

【样例 #3】

见附件中的 party/party3.inparty/party3.ans

该组样例满足测试点 11 的约束条件。

【样例 #4】

见附件中的 party/party4.inparty/party4.ans

该组样例满足测试点 232\sim3 的约束条件。

【样例 #5】

见附件中的 party/party5.inparty/party5.ans

该组样例满足测试点 454\sim5 的约束条件。

【样例 #6】

见附件中的 party/party6.inparty/party6.ans

该组样例满足测试点 696\sim9 的约束条件。

【样例 #7】

见附件中的 party/party7.inparty/party7.ans

该组样例满足测试点 101210\sim12 的约束条件。

【样例 #8】

见附件中的 party/party8.inparty/party8.ans

该组样例满足测试点 131513\sim15 的约束条件。

【样例 #9】

见附件中的 party/party9.inparty/party9.ans

该组样例满足测试点 202120\sim21 的约束条件。

【数据范围】

P=998244353P=998244353,对于所有测试数据,保证:1n3001\leq n\leq3001Mn(n1)21\leq M\leq\frac{n(n-1)}{2}1k<P1\leq k<P1l201\leq l\leq 201ci<P1\leq c_i<P(ui,vi)(u_i,v_i) 互不相同,给定图为有向无环图,在将有向边看成无向边的情况下,彩灯是连通的。

测试点编号 nn\leq kk\leq ll\leq 特殊性质
11 66 m10m\leq10ci=1c_i=1
232\sim3 300300 P1P-1 11
454\sim5 22
696\sim9 33
101210\sim12 2020 M=n1M=n-1 且有且仅有一个点入度为 00
131513\sim15 3030 1010
1616 150150 77
1717 1010
181918\sim19 1313
202120\sim21 300300 99
2222 1313
2323 1616
242524\sim25 2020