U138701.零昼回声:双相树谱

NOI/NOI+/CTSC

通过率:0%

时间限制:1.00s

内存限制:128MB

题目描述

题目背景

公元 3147 年,人类在“零昼带”发现了一种无法被传统坐标描述的物质结构——回声树

一棵回声树由若干枚记忆晶核与无向的相位通道组成。由于零昼带中的空间不会形成环,每两枚晶核之间都存在唯一的一条相位路径,因此整套结构恰好是一棵树。

每枚晶核 ii 同时携带两种彼此正交的相位:

  • 昼相强度 aia_i
  • 夜相强度 bib_i

当两枚不同晶核 u,vu,v 同时被激活时,它们会产生一种非常特殊的“双相交换干涉”。

若二者在回声树上的距离为 dd,则它们对第 dd 层回声频谱产生的贡献为

aubv+avbu.a_u b_v+a_v b_u.

研究人员并不关心某一对晶核,而是希望一次性恢复整棵回声树的完整距离频谱

对于每个可能的距离 dd,求所有距离恰好为 dd 的无序点对产生的总干涉强度。

由于零昼带中的相位数值极大,所有结果均在模数 998244353998244353 下计算。


题目描述

给定一棵包含 nn 个点的无根树。

ii 有两个整数权值 ai,bia_i,b_i

对于两个不同的点 u,vu,v,定义它们的双相干涉值为

I(u,v)=aubv+avbu.I(u,v)=a_u b_v+a_v b_u.

dist(u,v)\operatorname{dist}(u,v) 表示 u,vu,v 在树上的边数距离。

对于每个 d=1,2,,n1d=1,2,\ldots,n-1,定义

Rd=1u<vndist(u,v)=d(aubv+avbu).R_d= \sum_{\substack{1\le u<v\le n\\ \operatorname{dist}(u,v)=d}} \left(a_u b_v+a_v b_u\right).

请计算所有的

R1,R2,,Rn1.R_1,R_2,\ldots,R_{n-1}.

所有答案对 998244353998244353 取模。

输入格式

第一行一个整数 nn,表示回声树中的晶核数量。

第二行 nn 个整数

a1,a2,,an.a_1,a_2,\ldots,a_n.

第三行 nn 个整数

b1,b2,,bn.b_1,b_2,\ldots,b_n.

接下来 n1n-1 行,每行两个整数 u,vu,v,表示点 uu 与点 vv 之间存在一条无向边。

保证给出的图是一棵树。

输出格式

输出一行 n1n-1 个整数:

R1,R2,,Rn1.R_1,R_2,\ldots,R_{n-1}.

所有答案均对 998244353998244353 取模。

n=1n=1 时,输出一个空行。

输入输出样例

  • 输入#1

    5
    1 2 3 4 5
    5 4 3 2 1
    1 2
    1 3
    3 4
    3 5

    输出#1

    68 80 42 0

说明/提示

样例解释

距离为 11 的点对为

(1,2),(1,3),(3,4),(3,5).(1,2),(1,3),(3,4),(3,5).

它们的贡献分别为

1×4+2×5=14,1\times4+2\times5=14,

1×3+3×5=18,1\times3+3\times5=18,

3×2+4×3=18,3\times2+4\times3=18,

3×1+5×3=18.3\times1+5\times3=18.

因此

R1=14+18+18+18=68.R_1=14+18+18+18=68.

距离为 22 的所有点对总贡献为

R2=80.R_2=80.

距离为 33 的所有点对总贡献为

R3=42.R_3=42.

树中不存在距离为 44 的点对,因此

R4=0.R_4=0.


数据范围

对于所有测试数据:

1n2×105,1\le n\le 2\times10^5,

0ai,bi<998244353.0\le a_i,b_i<998244353.

保证输入图为一棵树。

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

首页