CFCF2167F.Tree, TREE!!!
普及/提高-
通过率:0%
AC君温馨提醒
该题目为【codeforces】题库的题目,您提交的代码将被提交至codeforces进行远程评测,并由ACGO抓取测评结果后进行展示。由于远程测评的测评机由其他平台提供,我们无法保证该服务的稳定性,若提交后无反应,请等待一段时间后再进行重试。
题目描述
树根可以变化,但树始终坚定如初——你的逻辑也应当如此坚韧。
Behruzbek 得到了一棵 n 个节点的树 $ ^{\text{∗}}$。对于一个选定的根节点 r †,Behruzbek 想要计算这棵树的可爱度(cuteness)。
考虑树上所有选取 k 个不同节点的集合。对于每一个这样的集合,计算在以 r 为根时,它们的最近公共祖先(LCA)。设 Sr 是所有通过上述操作得到的不重复节点的集合,那么,树的可爱度就是 ∣Sr∣,即集合中不同节点的个数。
在发现树的可爱度之后,Behruzbek 对树的 "kawaiiness" 感兴趣起来!"Kawaiiness" 定义为:
r=1∑n∣Sr∣=∣S1∣+∣S2∣+⋯+∣Sn∣
但现在 Behruzbek 感到困倦。请帮他计算这棵树的 "kawaiiness"!
∗ 一棵树是一个无环连通图。
† 一棵有根树是指定一个特殊节点为根结点的树。
输入格式
第一行包含测试用例个数 t(1≤t≤104)。
每个测试用例的第一行包含两个整数 n 和 k(2≤k≤n≤2⋅105)——树的节点数和要选择的不同节点个数。
每个测试用例接下来的 n−1 行描述树的结构。每行包含两个整数 u 和 v(1≤u,v≤n,u=v),表示存在一条连接节点 u 和 v 的边。保证所有边构成一棵树。
保证所有测试用例中 n 的总和不超过 2⋅105。
输出格式
对于每个测试用例,输出一个整数,即 r=1∑n∣Sr∣ 的值。
输入输出样例
输入#1
4 2 2 1 2 5 3 1 2 1 3 1 4 1 5 6 3 1 2 1 3 2 4 2 5 3 6 10 5 5 6 4 9 3 9 2 6 2 8 8 9 6 10 1 6 4 7
输出#1
2 9 17 35
说明/提示
令 f(i)=∣Si∣。
以第三个样例为例:
- 根是 1 时,只能得到 1 和 2 两个节点。例如,选择节点 4,5,6,LCA(4,5,6)=1;选择节点 2,4,5,LCA(2,4,5)=2。所以 f(1)=2。
- 根是 2 时,只能得到 1 和 2 两个节点。例如,选择节点 1,3,6,LCA(1,3,6)=1;选择节点 1,4,5,LCA(1,4,5)=2。所以 f(2)=2。
- 根是 3 时,f(3)=3。例如,选择节点 2,4,6,LCA(2,4,6)=3。
- 根是 4 时,f(4)=3。例如,选择节点 2,1,3,5,LCA(1,3,5)=2。
- 根是 5 时,f(5)=3。例如,选择节点 2,3,4,6,LCA(3,4,6)=2。
- 根是 6 时,f(6)=4。例如,选择节点 3,4,5,LCA(3,4,5)=2。
所以 2+2+3+3+3+4=17。