#P13320. [2025年队测]战争

    ID: 12504 传统题 1500ms 512MiB 尝试: 1 已通过: 1 难度: 8 上传者: 标签>CF2400树形DP组合数学动态规划多项式拓扑排序递归计数DP

[2025年队测]战争

题目背景

——高中三年是一场战争,而在这冰冷的战场上,我们相依取暖。

打开一个个漂流瓶,安又看到了数不清的祝福,以及随信附上的资料。

就像往常所做的一样,他记下了自己所需。他知道,在他观看的同时,系统已经将漂流瓶复制了若干份,送进了广阔的数据海中, 等待着下一个有缘的学生。

安很喜欢漂流瓶计划的主题:“知识与善意的指数式传递”。

被阅读得越多的漂流瓶便漂得越广,让本不相关的人们,在同样的年纪,并肩而战。

题目描述

题目内容和题目背景有部分区别,请注意。同时注意,你可以阅读形式化题意

安在自己的收件箱中看到了若干个同样种类的漂流瓶,上面的标题是:受战火波及,漂流瓶计划宣布破产。

惊讶的他发现今天是愚人节,而愚人节的玩笑是这样运作的:

ii 个人初始时会收到 aia_i 个完全一致的漂流瓶,安是第 11 个人。

每个人都可以观看若干个漂流瓶,每看一个就会给一个特定的人 pi<ip_i<i 送去 kk 个同样的漂流瓶。

最后有若干个漂流瓶送到安这里(可能为零),而他还没有看过任何一个漂流瓶(注意:虽然不合常理,但是他的信箱可能是空的 )。

经过这样的过程之后,安想要知道,每个人的信箱中,没有被看过的漂流瓶的可能状态数。定义两个状态不同,当且仅当存在一个 人信箱中没有被看过的漂流瓶的数量不同。答案对 998244353998244353 取模。

形式化题意

给定一棵 nn 个点以 11 为根的树,初始点有点权 aia_i

定义一次操作为选择一个 au>0a_u>0 的点,令 auau1,afauafau+ka_u\gets a_u-1,a_{fa_u}\gets a_{fa_u}+k。求任意次操作后可能得到的序列 aa 的数量。

998244353998244353 取模。

输入格式

第一行两个正整数 tid,Ttid,T 表示测试数据编号和测试数据组数。

对于每组数据:第一行输入两个数 n,kn,k 表示人数的复制的倍数。

第二行输入 n1n-1 个数,第 i1i-1 个数表示 pip_i

第三行输入 nn 个数表示初始漂流瓶个数 aa

输出格式

对于每组数据,输出一行一个数表示答案对 998244353998244353 取模后的结果。

输入输出样例 #1

输入 #1

0 2
3 2
1 1
0 1 1
3 998244352
1 2
0 1 1

输出 #1

4
3

说明/提示

【样例解释 #1】

对于第一组数据,第二和第三个人有没有看过漂流瓶决定了最终状态,因此答案为 44

对于第二组数据,状态数为 998244356998244356


VVaia_i 的最大值。测试数据满足:

数据点编号 nn\le kk\le VV 特殊性质
1,21,2 1010 11 1010
33 998244352998244352 10910^9 A
44 00 C
55 998244352998244352
6,76,7 5050 11 1010
88 10910^9
99 998244352998244352 1010 A
1010 10910^9
1111 B
1212 1010
1313 10910^9
14,1514,15 3×1023\times10^2 11 1010
16,1716,17 10910^9
1818 998244352998244352 A
1919 00 C
2020 998244352998244352 B
21,2221,22 1010
23,24,2523,24,25 10910^9
  • 特殊性质 A:保证 pi=ip_i=i
  • 特殊性质 B:保证 pi=1p_i=1
  • 特殊性质 C:保证 k=0k=0
  • 对于所有数据,满足 T=3,n3×102,k998244352,V109T=3,n\le3\times10^2,k\le998244352,V\le10^9