#ATarc150d. [ARC150D] Removing Gacha

[ARC150D] Removing Gacha

题目描述

有一棵有 NN 个顶点的有根树,顶点编号为 11NN。顶点 11 是这棵树的根,对于每个顶点 i (2i)i\ (2\leq i),其父节点为 pip_i

每个顶点有白色和黑色两种颜色。初始时所有顶点都是白色。

在这棵有根树中,如果从根节点 11 到顶点 ii 的唯一简单路径上的所有顶点(包括 11ii)都是黑色,则称顶点 ii 为“好顶点”。否则称为“坏顶点”。

你需要重复进行如下操作,直到所有顶点都变为黑色为止:从所有“坏顶点”中等概率随机选取一个顶点,并将其涂成黑色。

请你求出将所有顶点都变为黑色所需操作次数的期望值,并对 998244353998244353 取模。

期望值 mod 998244353\bmod\ 998244353 的定义:可以证明,所求的期望值一定是有理数,并且在本题的约束下,若将其表示为最简分数 PQ\frac{P}{Q},则 Q≢0(mod998244353)Q\not\equiv 0\pmod{998244353}。因此,存在唯一的整数 RR 满足 $R\times Q\equiv P\pmod{998244353},\ 0\leq R<998244353$。请输出这个 RR

输入格式

输入为以下格式,从标准输入读取。

NN p2p_2 p3p_3 \dots pNp_{N}

输出格式

请输出答案。

样例 1

输入

4
1 1 3

输出

831870300

样例 2

输入

15
1 2 1 1 4 5 3 3 5 10 3 6 3 13

输出

515759610

说明/提示

数据范围

  • 2N2×1052\leq N\leq 2\times 10^5
  • 1pi<i1\leq p_i<i
  • 输入的所有值均为整数

样例解释 1

例如,假设第 1, 2, 31,\ 2,\ 3 次操作依次选择了顶点 1, 2, 41,\ 2,\ 4。此时,顶点 1, 21,\ 2 是“好顶点”,但顶点 44 由于其祖先顶点 33 仍为白色,所以是“坏顶点”。因此在第 44 次操作时,需要从顶点 3, 43,\ 4 中等概率随机选择一个。将所有顶点变为黑色所需操作次数的期望值为 356\displaystyle\frac{35}{6}

由 ChatGPT 4.1 翻译