#P15073. [2026省选联测季积晓淆

    ID: 14289 传统题 1500ms 1024MiB 尝试: 2 已通过: 1 难度: 9 上传者: 标签>CF2800概率DP树形DP数学FFT数论生成函数动态规划

[2026省选联测季积晓淆

题目描述

有一棵 nn 个节点的树,你将从任意一个点出发开始随机游走。

具体来说,在点 uu 的每个单位时间内你将会有 pup_u 的概率留在原地,有 1pu1−p_u 的概率等概率的向相邻的点移动,直到移动到 11 号点才停下。

现在询问从每个点出发直至停下,所花费的时间的 kk 次方的期望。

可以证明,答案可以被表示成 q×p1q × p^{-1}的形式,你需要输出一个非负整数 ansans, 使得 ansq×p1(mod998244353)ans ≡ q × p^{−1}(\mod 998244353), 保证 pp 将不会是 998244353998244353 的倍数。

输入格式

第一行两个整数 nnkk, 含义如题目所示。

接下来 n1n − 1 行,每行两个整数 u,vu, v,代表一条树边。

接下来一行 n1n − 1 个整数, 第 iipi+1p^{'}_{i+1}pi+1=pi+1×106p_{i+1} =p^{'}_{i+1} × 10^{-6} 代表留在原地的概率。

输出格式

输出 n1n − 1 行,第 ii 输出从 i+1i + 1 号点出发直至停下,所花费的时间的 kk 次方的期望。

样例

样例输入1

3 1
1 2
2 3
0 0

样例输出1

3
4

样例输入2

3 1
1 2
2 3
500000 500000

样例输出2

6
8

样例输入3

3 2
1 2
2 3
500000 500000

样例输出3

74
104

数据范围与提示

对于 15%15\% 的数据,保证 n10n \leq 10, k10k \leq 10

对于 30%30\% 的数据,保证 n50n \leq 50, k50k \leq 50

对于 50%50\% 的数据,保证 n1000n \leq 1000, k100k \leq 100

对于另外 5%5\% 的数据,保证 k=0k = 0

对于另外 15%15\% 的数据,保证 k=1k = 1

对于另外 10%10\% 的数据,保证 pi=0p_i = 0

对于 95%95\% 的数据,保证 k1000k \leq 1000

对于 100%100\% 的数据,保证 nk<=106nk <= 10^6, 1n1051 \leq n \leq 10^5,0k1050 \leq k \leq 10^5, 0pi<1060 \leq p_i < 10^6