AT_arc150_d.[ARC150D] Removing Gacha

省选/NOI-

通过率:0%

AC君温馨提醒

该题目为【atcoder】题库的题目,您提交的代码将被提交至atcoder进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。

题目描述

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

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

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

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

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

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

输入格式

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

NN p2p_2 p3p_3 …\dots pNp_{N}

输出格式

请输出答案。

输入输出样例

  • 输入#1

    4
    1 1 3

    输出#1

    831870300
  • 输入#2

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

    输出#2

    515759610

说明/提示

数据范围

  • 2≤N≤2×1052\leq N\leq 2\times 10^5
  • 1≤pi<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 翻译

输入解题思路,AI测评打分。不知道怎么写?

首页