
树路径复合和(Tree Path Composite Sum)
问题描述
给你一棵含 N 个顶点的树,以及以下输入:
- 整数序列 a0,a1,…,aN−1;
- N−1 条边,第 i 条边连接顶点 ui 和 vi,并关联参数 bi,ci;
- 对每个边索引 e(0≤e≤N−2),定义线性函数 fe(x)=bex+ce。
对任意顶点对 (x,y),设 e0,e1,…,ek 为从 x 到 y 的唯一简单路径上的边(按顺序),定义复合函数
$$P(x, y) = f_{e_k}(f_{e_{k-1}}(\cdots f_{e_0}(a_y)\cdots)).$$
请对每个顶点 x,计算
qx=y=0∑N−1P(x,y)mod998244353.
约束条件
- 所有输入均为整数。
- 1≤N≤2×105
- 0≤ui,vi≤N−1
- 0≤ai<998244353
- 1≤bi<998244353
- 0≤ci<998244353
输入格式
N
a0 a1 ⋯ aN−1
u0 v0 b0 c0
u1 v1 b1 c1
:
uN−2 vN−2 bN−2 cN−2
输出格式
q0 q1 ⋯ qN−1
8
1 2 3 4 5 6 7 8
0 1 10 1
1 2 10 1
0 3 10 1
0 4 10 0
0 5 10 1
5 6 10 0
6 7 10 1
9287 89589 895590 92471 92375 5131 42598 425185
3
1 100 10000
0 1 100000 0
1 2 100000 0
881938226 1855747 27566470