AT_abc376_g [ABC376G] Treasure Hunting
题目描述
有一棵有 $N+1$ 个顶点的有根树,顶点编号为 $0$ 到 $N$。顶点 $0$ 是根,顶点 $i$ 的父节点是顶点 $p_i$。
在顶点 $1$、顶点 $2$、$\dots$、顶点 $N$ 中的某一个顶点藏有宝物。顶点 $i$ 藏有宝物的概率为 $\frac{a_i}{\sum_{j=1}^N a_j}$。
此外,每个顶点有“已探索”和“未探索”两种状态。初始时,顶点 $0$ 为已探索,其余顶点均为未探索。
你需要进行如下操作,直到藏有宝物的顶点变为已探索:
- 选择一个父节点已探索的未探索顶点,将其状态变为已探索。
请在操作次数的期望值最小的情况下,求出操作次数的期望值,并对 $998244353$ 取模。
给定 $T$ 组测试数据,请分别输出每组的答案。
期望值 $\bmod\ 998244353$ 的定义:
可以证明,所求期望值一定是有理数。在本题的约束下,若将其表示为最简分数 $\frac{P}{Q}$,则 $Q \not\equiv 0 \pmod{998244353}$ 也成立。此时,存在唯一的整数 $R$ 满足 $R \times Q \equiv P \pmod{998244353},\ 0 \leq R < 998244353$。请输出这个 $R$。
输入格式
输入按如下格式从标准输入给出,其中 $\mathrm{case}_i$ 表示第 $i$ 个测试用例。
> $T$
> $\mathrm{case}_1$
> $\mathrm{case}_2$
> $\vdots$
> $\mathrm{case}_T$
每个测试用例格式如下:
> $N$
> $p_1\ p_2\ \dots\ p_N$
> $a_1\ a_2\ \dots\ a_N$
输出格式
输出 $T$ 行,第 $i$ 行输出第 $i$ 个测试用例的答案。
说明/提示
### 数据范围
- $1 \leq T \leq 2 \times 10^5$
- $1 \leq N \leq 2 \times 10^5$
- $0 \leq p_i < i$
- $1 \leq a_i$
- $\sum_{i=1}^N a_i \leq 10^8$
- 所有测试用例中 $N$ 的总和不超过 $2 \times 10^5$
- 输入的所有数均为整数
### 样例解释 1
第 $1$ 个测试用例中,操作次数的期望值为 $\frac{13}{6}$。
由 ChatGPT 4.1 翻译