AT_abc376_g.[ABC376G] Treasure Hunting

NOI/NOI+/CTSC

通过率:0%

AC君温馨提醒

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

题目描述

有一棵有 N+1N+1 个顶点的有根树,顶点编号为 00 到 NN。顶点 00 是根,顶点 ii 的父节点是顶点 pip_i。
在顶点 11、顶点 22、…\dots、顶点 NN 中的某一个顶点藏有宝物。顶点 ii 藏有宝物的概率为 ai∑j=1Naj\frac{a_i}{\sum_{j=1}^N a_j}。
此外,每个顶点有“已探索”和“未探索”两种状态。初始时,顶点 00 为已探索,其余顶点均为未探索。
你需要进行如下操作,直到藏有宝物的顶点变为已探索:

  • 选择一个父节点已探索的未探索顶点,将其状态变为已探索。

请在操作次数的期望值最小的情况下,求出操作次数的期望值,并对 998244353998244353 取模。

给定 TT 组测试数据,请分别输出每组的答案。

期望值  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。

输入格式

输入按如下格式从标准输入给出,其中 casei\mathrm{case}_i 表示第 ii 个测试用例。

TT
case1\mathrm{case}_1
case2\mathrm{case}_2
⋮\vdots
caseT\mathrm{case}_T

每个测试用例格式如下:

NN
p1 p2 … pNp_1\ p_2\ \dots\ p_N
a1 a2 … aNa_1\ a_2\ \dots\ a_N

输出格式

输出 TT 行,第 ii 行输出第 ii 个测试用例的答案。

输入输出样例

  • 输入#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×1051 \leq T \leq 2 \times 10^5
  • 1≤N≤2×1051 \leq N \leq 2 \times 10^5
  • 0≤pi<i0 \leq p_i < i
  • 1≤ai1 \leq a_i
  • ∑i=1Nai≤108\sum_{i=1}^N a_i \leq 10^8
  • 所有测试用例中 NN 的总和不超过 2×1052 \times 10^5
  • 输入的所有数均为整数

样例解释 1

第 11 个测试用例中,操作次数的期望值为 136\frac{13}{6}。

由 ChatGPT 4.1 翻译

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

首页