D2. LuoTianyi and the Floating Islands (Hard Version)

Problem - D2 - Codeforces

题目大意：有一棵n个点的树，随机使其中的k个点变成关键点，问到所有关键点的简单路径和最小的点的期望数量

1<=k<=n<=2e5;

思路：当点数为奇数的时候，只有中间的那个点是符合要求的点，因为如果向左或向右移动一步，到k/2+1个点的距离会分别+1，到k/2-1个点的距离会分别-1，所以总距离+1，那么当n位奇数时，期望点数只有1。当n为偶数时，同理也只有中间两个点之间的这些点满足条件，如果在这些点中挪动，那么到k/2个点的距离分别+1，到另k/2个点的距离分别-1，总距离不变，如果移出这个范围，到k/2+1个点的距离会分别+1，到k/2-1个点的距离会分别-1，所以总距离+1，所以对于每一个点，可行点数v就是一半子树放k/2个点，另一半子树放k/2个点，等于C(k/2,子树大小)*C(k/2,另一边的子树大小)总的放点t的方案数是C(k,n)，然后因为我们遍历时是左闭右开的遍历，所以答案是1+v/t

//#include<__msvc_all_public_headers.hpp>
#include<bits/stdc++.h>
using namespace std;
const int N = 2e5 + 5;
typedef long long ll;
const ll MOD = 1e9 + 7;
vector<int>g[N];
bool vis[N];
ll s[N];
ll n, k;
ll ans = 0;
ll fac[N];
ll qpow(ll a, ll b)
{//快速幂求逆元
	ll ret = 1;
	while (b)
	{
		if (b & 1)
		{
			ret = ret * a % MOD;
		}
		b >>= 1;
		a = a * a % MOD;
	}
	return ret ;
}
ll C(ll a, ll b)
{//求组合数
	if (a > b)
		return 0;
	return (fac[b] * qpow(fac[a] * fac[b - a] % MOD, MOD - 2) % MOD);//b!/(a!*(b-a)!)
}
void dfs(int u)
{
	vis[u] = 1;
	s[u] = 1;
	for (int i = 0; i < g[u].size(); i++)
	{
		int v = g[u][i];
		if (!vis[v])
		{
			dfs(v);
			s[u] += s[v];//记录子树大小
		}
	}
	ans = (ans + C(k / 2, s[u]) * C(k / 2,n - s[u]) % MOD) % MOD;//维护每个点的贡献
}
int main()
{
	cin.tie(0);
	cout.tie(0);
	ios::sync_with_stdio(false);
	cin >> n >> k;
	fac[0] = 1;
	for (int i = 1; i <= n; i++)
	{
		fac[i] = fac[i - 1] * i % MOD;//预处理阶乘数组
	}
	for (int i = 1; i <= n - 1; i++)
	{
		int u, v;
		cin >> u >> v;
		g[u].push_back(v);
		g[v].push_back(u);
	}
	if (k & 1)
	{
		cout << 1 << endl;
	}
	else
	{
		dfs(1);
		cout << (1 + ans * qpow(C(k, n), MOD - 2) % MOD) % MOD << endl;
	}
	return 0;
}