luogu#P8981. 「DROI」Round 1 距离

    ID: 12966 远端评测题 1000~3000ms 512MiB 尝试: 0 已通过: 0 难度: 5 上传者: 标签>树形数据结构洛谷原创O2优化树形 dp树的直径

「DROI」Round 1 距离

题目背景

没有什么距离是无法跨越的。

题目描述

定义一棵树 GG 上两点 u,vu,v 之间的距离 dis(u,v)\operatorname{dis}(u,v) 为两点之间点的数量。

若对于树上两点 u,vu,v,满足 $\forall x \in G,\operatorname{dis}(u,x) \leq \operatorname{dis}(u,v)$ $\operatorname{dis}(v,x) \leq \operatorname{dis}(u,v)$,那么我们称无序点对 (u,v)(u,v)极远点对

同时,树 GG 上一点 xx 的权值 vxv_x 定义为:满足两点间最短路径经过 xx 的极远点对的数量。

现给定树 GG,求 xGvxk\sum\limits_{x \in G}{v_x^k}998244353998244353 取模的值,其中 kk 是给定的常数,且 k[1,2]k \in [1,2]

输入格式

第一行两个数 n,kn,k,分别表示树 GG 的点数以及给定的常数。

接下来 n1n-1 行每行两个整数 u,vu,v,表示点 uu 和点 vv 之间有一条边。

输出格式

输出一行一个整数,表示答案。

2 1
1 2

2
5 2
1 2
1 3
4 1
5 1

72

提示

样例解释 #1

(1,2)(1,2) 为极远点对,所以 11 号和 22 号点点权均为 1111+11=21^1 + 1^1 =2


样例解释 #2

极远点对有 (2,3),(2,4),(2,5),(3,4),(3,5),(4,5)(2,3),(2,4),(2,5),(3,4),(3,5),(4,5),故答案为 4×32+62=724 \times 3^2 + 6^2 = 72


数据范围

测试点编号 11 22 33 454 \sim 5 66 77 898 \sim 9 1010
nn 300300 20002000 10510^5 5×1065 \times 10^6 10510^5 5×1065 \times 10^6
kk 11 22 11 22 11 22

对于 100%100\% 的数据,满足 n5×106n \leq 5 \times 10^61k21 \leq k \leq 2

本题输入量较大,请用较快的输入方法。