时间限制 : 20000 MS 空间限制 : 565536 KB
评测说明 : 2s,512m
master 对树上的求和非常感兴趣。他生成了一棵有根树,并且希望多次询问这棵树上一段路径上所有节点深度的k次方和,而且每次的k 可能是不同的。此处节点深度的定义是这个节点到根的路径上的边数。他把这个问题交给了pupil,但pupil 并不会这么复杂的操作,你能帮他解决吗?
第一行包含一个正整数n ,表示树的节点数。
之后n-1 行每行两个空格隔开的正整数i,j ,表示树上的一条连接点i 和点j 的边。
之后一行一个正整数m ,表示询问的数量。
之后每行三个空格隔开的正整数i,j,k ,表示询问从点i 到点j 的路径上所有节点深度的k 次方和。由于这个结果可能非常大,输出其对998244353 取模的结果。
树的节点从1 开始标号,其中1 号节点为树的根。
对于每组数据输出一行一个正整数表示取模后的结果。
5
1 2
1 3
2 4
2 5
2
1 4 5
5 4 45
33
503245989
对于30%的数据,1≤n,m≤100;
对于60%的数据,1≤n,m≤1000;
对于100%的数据,1≤n,m≤300000,1≤k≤50。
样例解释
以下用d(i) 表示第i 个节点的深度。
对于样例中的树,有d(1)=0,d(2)=1,d(3)=1,d(4)=2,d(5)=2。
因此第一个询问答案为(2^5 + 1^5 + 0^5) mod 998244353 = 33
第二个询问答案为(2^45 + 1^45 + 2^45) mod 998244353 = 503245989。
【分析】
我们首先想想,树上两点的距离是怎么求的呢?先用DFS预处理,遍历树上每一个点,记下i号点与根节点的距离,记做Dis[i][1]。而点i,j间的距离即可表示为Dis[i][j] = Dis[i][1] + Dis[j][1] - 2 * Dis[(Lca(i, j)][1]。
于是,这道题我们维护一个pre[ ]数组,pre[u][k]表示以u为节点,k为题目给定的权值,在这棵树中的前缀和。那么,对于样例的第二组数据:pre[4][45]表示从4到根节点1的答案。那么我们计算4与5之间的总
和自然就是pre[4][k] + pre[5][k] - pre[Lca(4, 5)][k] - pre[F[Lca(4, 5)][0]][k]。
【标程】
手机扫一扫
移动阅读更方便
你可能感兴趣的文章