#CF2167F. 树,树!!!

    ID: 6984 传统题 1000ms 256MiB 尝试: 1 已通过: 1 难度: 10 上传者: 标签>深度优先搜索动态规划数学CodeforcesCodeforces Round 1062(Div4)Div4FCF2167F1600

树,树!!!

题目描述

给定一棵有 nn 个节点的树,以及一个整数 kk

对于一个选定的根 rr,考虑树以 rr 为根时的形态。现在从树中选择任意 kk 个互不相同的节点,求这 kk 个节点的最近公共祖先 LCA。

SrS_r 表示:在根为 rr 时,所有可能选出的 kk 个节点集合对应的 LCA 所形成的不同节点集合。

根为 rr 时,这棵树的可爱度定义为 Sr|S_r|

请计算所有根的可爱度之和:

r=1nSr\sum_{r=1}^{n} |S_r|

输入格式

第一行一个整数 tt,表示测试组数。

每组测试数据第一行包含两个整数 n,kn,k

接下来 n1n-1 行,每行两个整数 u,vu,v,表示树上的一条边。

保证输入的边构成一棵树,且所有测试数据的 nn 之和不超过 21052\cdot 10^5

输出格式

对于每组测试数据,输出一个整数,表示所有根的可爱度之和。

样例

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
2
9
17
35

样例说明

f(i)=Sif(i)=|S_i|

第三组测试数据中:

  • 根为 11 时,只能得到节点 1122,例如 LCA(4,5,6)=1LCA(4,5,6)=1LCA(2,4,5)=2LCA(2,4,5)=2,所以 f(1)=2f(1)=2
  • 根为 22 时,只能得到节点 1122,例如 LCA(1,3,6)=1LCA(1,3,6)=1LCA(1,4,5)=2LCA(1,4,5)=2,所以 f(2)=2f(2)=2
  • 根为 33 时,f(3)=3f(3)=3,例如选择节点 2,4,62,4,6 可以得到 LCA(2,4,6)=3LCA(2,4,6)=3
  • 根为 44 时,f(4)=3f(4)=3,例如选择节点 1,3,51,3,5 可以得到节点 22
  • 根为 55 时,f(5)=3f(5)=3,例如选择节点 3,4,63,4,6 可以得到节点 22
  • 根为 66 时,f(6)=4f(6)=4,例如选择节点 3,4,53,4,5 可以得到节点 22

因此答案为 2+2+3+3+3+4=172+2+3+3+3+4=17

数据范围

  • 1t1041 \le t \le 10^4
  • 2kn21052 \le k \le n \le 2\cdot 10^5
  • 1u,vn1 \le u,v \le n
  • 所有测试数据的 nn 之和不超过 21052\cdot 10^5

来源

Codeforces Round 1062 (Div. 4), Problem F - Tree, TREE!!!