AT_abc376_g.[ABC376G] Treasure Hunting
NOI/NOI+/CTSC
通过率:0%
AC君温馨提醒
该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
有一棵有 N+1 个顶点的有根树,顶点编号为 0 到 N。顶点 0 是根,顶点 i 的父节点是顶点 pi。
在顶点 1、顶点 2、…、顶点 N 中的某一个顶点藏有宝物。顶点 i 藏有宝物的概率为 ∑j=1Najai。
此外,每个顶点有“已探索”和“未探索”两种状态。初始时,顶点 0 为已探索,其余顶点均为未探索。
你需要进行如下操作,直到藏有宝物的顶点变为已探索:
- 选择一个父节点已探索的未探索顶点,将其状态变为已探索。
请在操作次数的期望值最小的情况下,求出操作次数的期望值,并对 998244353 取模。
给定 T 组测试数据,请分别输出每组的答案。
期望值 mod 998244353 的定义:
可以证明,所求期望值一定是有理数。在本题的约束下,若将其表示为最简分数 QP,则 Q≡0(mod998244353) 也成立。此时,存在唯一的整数 R 满足 R×Q≡P(mod998244353), 0≤R<998244353。请输出这个 R。
输入格式
输入按如下格式从标准输入给出,其中 casei 表示第 i 个测试用例。
T
case1
case2
⋮
caseT
每个测试用例格式如下:
N
p1 p2 … pN
a1 a2 … aN
输出格式
输出 T 行,第 i 行输出第 i 个测试用例的答案。
输入输出样例
输入#1
3 3 0 0 1 1 2 3 5 0 1 0 0 0 8 6 5 1 7 10 0 1 1 3 3 1 4 7 5 4 43 39 79 48 92 90 76 30 16 30
输出#1
166374061 295776107 680203339
说明/提示
数据范围
- 1≤T≤2×105
- 1≤N≤2×105
- 0≤pi<i
- 1≤ai
- ∑i=1Nai≤108
- 所有测试用例中 N 的总和不超过 2×105
- 输入的所有数均为整数
样例解释 1
第 1 个测试用例中,操作次数的期望值为 613。
由 ChatGPT 4.1 翻译
输入解题思路,AI测评打分。不知道怎么写?