时间限制 : 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]。

【标程】

最新文章

  1. C#开发微信门户及应用(7)-微信多客服功能及开发集成
  2. Python中两种处理错误方法的比较
  3. [HTML/HTML5]3 页面结构
  4. JQuery按回车提交数据
  5. Jenkins环境拓扑及部署流程
  6. 使一个div始终显示在页面中间
  7. Metro各种流转换
  8. 团队交流合作简单解决方案:TeamViewer远程控制&会议演示 + HyperCam屏幕录制(免费)
  9. C++类编程(一)const的使用
  10. python中用filter求素数
  11. 暴走Python之运算符与条件语句
  12. mysql远程连接 Host * is not allowed to connect to this MySQL server(第一次配置好lnmp环境)
  13. Python之路,第十一篇:Python入门与基础11
  14. Git的一些常用命令
  15. CF#235E. Number Challenge
  16. JNI学习笔记_C调用Java
  17. nginx alias
  18. [IOI2011]Race 点分治
  19. 商业化IM 客户端接口设计分析
  20. 自己动手实现arm函数栈帧回溯【转】

热门文章

  1. git add的各种情况分类
  2. JavaScript(js)函数声明与函数表达式的区别
  3. NoVNC API 文档翻译
  4. Simulink仿真入门到精通(十四) Simulink自定义环境
  5. vue中moment.js的使用
  6. mongo复制数据库和表
  7. 「CSS」常见的清除浮动方法
  8. 【分布式锁】01-使用Redisson实现可重入分布式锁原理
  9. ssh 公钥 下载选择的时候 下拉选择 ssh 然后 git clone
  10. maven中的pom配置文件一——spring,mybatis,oracle,jstl,json,文件上传