51Nod - 1677（数学+算贡献）

题目链接
给定一棵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取模。
Sample Input
3 2
1 2
1 3
Sample Output
4
Sponsor
思路：首先对于每条边，总的方案是C（n，k）但是对于一条边x来讲，假设x左边有size个孩子，右边就是n-size个，要想x产生贡献，必须左边选一点，右边选一点，只选一边是不可能产生贡献的，所以一条边的贡献就是总的减去两边的，也就是C(n,k)-C(size,k)-C(n-size,k)。

#include<bits/stdc++.h>
using namespace std;
const int maxn=1e5+5;
typedef long long ll;
const int mod=1e9+7;
ll res,f[maxn];
vector<int>g[maxn];
int n,k,size[maxn];
ll quick(ll a,ll b)
{
	ll ans=1;
	a%=mod;
	while(b!=0)
	{
		if(b&1) ans=(ans*a)%mod;
		b>>=1;
		a=(a*a)%mod; 
	}
	return ans;
}
ll C(int n,int m)
{
	if(n<m) return 0;
	return f[n]*quick(f[m],mod-2)%mod*quick(f[n-m],mod-2)%mod;
}
void dfs(int x,int fa)
{
	size[x]=1;
	for(int to:g[x])
	{
		if(to==fa) continue;
		dfs(to,x);
		res=(res+C(n,k)-C(size[to],k)-C(n-size[to],k)%mod+mod)%mod;
		size[x]+=size[to];
	}
}
int main()
{
	scanf("%d%d",&n,&k);
	for(int i=1,u,v;i<n;++i)
	{
		scanf("%d%d",&u,&v);
		g[u].push_back(v);
		g[v].push_back(u);
	}
	f[0]=1;
	for(int i=1;i<=n;++i) f[i]=(f[i-1]*i)%mod;
	dfs(1,-1);
	printf("%lld\n",res);
}