给定一棵n个节点的树,从1到n标号。选择k个点,你需要选择一些边使得这k个点通过选择的边联通,目标是使得选择的边数最少。
现需要计算对于所有选择k个点的情况最小选择边数的总和为多少。
样例解释:
一共有三种可能:(下列配图蓝色点表示选择的点,红色边表示最优方案中的边)
选择点{1,2}:至少要选择第一条边使得1和2联通。
选择点{1,3}:至少要选择第二条边使得1和3联通。
选择点{2,3}:两条边都要选择才能使2和3联通。
Input
第一行两个数n,k(1<=k<=n<=100000)
接下来n-1行,每行两个数x,y描述一条边(1<=x,y<=n)
Output
一个数,答案对1,000,000,007取模。
Input示例
3 2
1 2
1 3
Output示例
4
思路:
假设一条边左端点相连的节点个数为x, 右端点相连的节点个数为y,则通过这条边的总数为C(n, k) - C(x, k) - C(y, k);
将所有边的总数相加即可。
计算组合时需要用到逆元。
#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;
typedef long long ll;
const ll MOD = 1e9 + 7;
const ll MAXN = 1e5 + 5;
vector<int > edge[MAXN];
ll ans;
ll n, k;
bool vis[MAXN];
ll fac[MAXN];
ll inv_of_fac[MAXN];
ll qPow(ll x, ll n)
{
ll ret = 1;
while (n > 0)
{
if (n & 1)
{
ret = ret * x % MOD;
}
x = x * x % MOD;
n >>= 1;
}
return ret;
}
void init()
{
fac[1] = 1;
for (int i = 2; i <= MAXN - 2; i++)
{
fac[i] = fac[i - 1] * i % MOD;
}
inv_of_fac[MAXN - 2] = qPow(fac[MAXN - 2], MOD - 2);
for (int i = MAXN - 3; i >= 0; i--)
{
inv_of_fac[i] = inv_of_fac[i + 1] * (i + 1) % MOD;
}
}
ll C(ll a, ll b)
{
if (b > a)
{
return 0;
}
if (b == 0)
{
return 1;
}
return fac[a] * inv_of_fac[b] % MOD * inv_of_fac[a - b] % MOD;
}
ll dfs(int node)
{
vis[node] = true;
ll count = 1;
for (ll i = 0; i < edge[node].size(); i++)
{
ll peer = edge[node][i];
if (!vis[peer])
{
ll tmp = dfs(peer);
ans = (ans + (C(n, k) % MOD - C(tmp, k) % MOD - C(n - tmp, k) % MOD) % MOD + MOD) % MOD;
count += tmp;
}
}
return count;
}
int main()
{
while (~scanf("%lld%lld", &n, &k))
{
init();
memset(vis, 0, sizeof(vis));
for (int i = 1; i <= n; i++)
{
edge[i].clear();
}
for (int i = 1; i <= n - 1; i++)
{
int x, y;
scanf("%d %d", &x, &y);
edge[x].push_back(y);
edge[y].push_back(x);
}
ans = 0;
dfs(1);
printf("%lld\n", (ans + MOD) % MOD);
}
return 0;
}