51nod 1677 treecnt【Dfs+思维+数论】好题

给定一棵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

思路：

1、（窝不会Cnk啊）首先感谢http://www.cnblogs.com/Just--Do--It/p/6103326.html

来自zzwy的Cnk的板子....数论渣的一逼，借鉴了大牛的代码，鸣谢。

2、首先考虑这样一个问题，对于一个边，其贡献度为多少？对应看这样一个图（图中左侧共X个点，右侧共Y个点，包括边上的点）：

我们需要每次找K个点，那么对应我们可能有三种取法：

①在这条边的两侧都有点被取了出来

②只在左侧选K个点

③只在右侧选K个点

对应最后两种情况是没有贡献度的.那么对应去掉即可。

那么对应这条边的贡献度为：C（n，k）-C（x，k）-C（y，k）；

3、那么Dfs处理完，计算总和，注意取模。

Ac代码：

#include<stdio.h>
#include<string.h>
#include<vector>
using namespace std;
#define LL __int64
#define MOD 1000000007ll
const LL mod = 1000000007;
const LL N = 100000+5;
const LL M=1e5+3;
vector<LL >mp[100500];
LL ans;
LL n,k;
LL vis[100500];
LL fac[100005];            //阶乘
LL inv_of_fac[100005];        //阶乘的逆元
LL qpow(LL x,LL n)
{
    LL ret=1;
    for(; n; n>>=1)
    {
        if(n&1) ret=ret*x%mod;
        x=x*x%mod;
    }
    return ret;
}
void init()
{
    fac[1]=1;
    for(int i=2; i<=M; i++)
        fac[i]=fac[i-1]*i%mod;
    inv_of_fac[M]=qpow(fac[M],mod-2);
    for(int i=M-1; 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;
}
//(C(k,n)-C(k,cont)-C(k,n-cont)+MOD)%MOD;
LL Dfs(int u)
{
    vis[u]=1;
    LL cont=1;
    for(LL int i=0;i<mp[u].size();i++)
    {
        LL v=mp[u][i];
        if(vis[v]==0)
        {
            LL tmp=Dfs(v);
            ans=(ans+(C(n,k)%MOD-C(tmp,k)%MOD-C(n-tmp,k)%MOD)%MOD+MOD)%MOD;
            cont+=tmp;
        }
    }
    return cont;
}
int main()
{
    while(~scanf("%I64d%I64d",&n,&k))
    {
        init();
        memset(vis,0,sizeof(vis));
        for(LL i=1;i<=n;i++)mp[i].clear();
        for(LL i=1;i<=n-1;i++)
        {
            LL x,y;
            scanf("%I64d%I64d",&x,&y);
            mp[x].push_back(y);
            mp[y].push_back(x);
        }
        ans=0;
        Dfs(1);
        printf("%I64d\n",(ans+MOD)%MOD);
    }
    //prLLf("%I64d\n",C(5,2));
}