51Nod-1677 treecent(计数)

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

题解：对于每一条边，如果左边有a个点，右边有b个点，那么如果这条边被选中，那么左边至少有一个点被选中，右边 至少有一个点被选中，因此对答案的贡献为C(n,k)-C(a,k)-C(b,k),其中C(a,k)表示k个点全部在左边的方案数，C(b,k)表示k个点全部在右边的方案数

#include <cstdio>
#include<string.h>
#include <algorithm>
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
typedef long long LL;
const int MX = 1e5 + 5;
const LL mod = 1e9 + 7;
struct Ecge{
    int v,nxt;
}edge[MX*2];
int n,k,tot,head[MX],c[MX];
void add(int u,int v){
    edge[tot].v=v;
    edge[tot].nxt=head[u];
    head[u]=tot++;
}
int dfs(int u,int fa){
    int cnt=1;
    for(int i=head[u];~i;i=edge[i].nxt){
        int v=edge[i].v;
        if(v==fa) continue;
        c[i/2]=dfs(v,u);
        cnt+=c[i/2];
    }
    return cnt;
}

LL F[MX], invF[MX];
LL power(LL a, LL b) {
    LL ret = 1;
    while(b) {
        if(b & 1) ret = (ret * a) % mod;
        a = (a * a) % mod;
        b >>= 1;
    }
    return ret;
}
void init() {
    memset(head,-1,sizeof(head));
    tot=0;
    F[0] = 1;
    for(int i = 1; i < MX; i++){
        F[i] = (F[i - 1] * i) % mod;
}
    invF[MX - 1] = power(F[MX - 1], mod - 2);
    for(int i = MX - 2; i >= 0; i--) {
        invF[i] = invF[i + 1] * (i + 1) % mod; //invF[i]*i!=1,invF[i+1]*i!*(i+1)=1
    }
}

LL C(int n, int m) {
    if(n < 0 || m < 0 || m > n) return 0;
    if(m == 0 || m == n)    return 1;
    return F[n] * invF[n - m] % mod * invF[m] % mod;
}

int main(){
    while(~scanf("%d%d",&n,&k)){
        init();
        for(int i=1;i<n;i++){
            int u,v;
            scanf("%d%d",&u,&v);
            add(u,v);
            add(v,u);
        }
        dfs(1,0);
        LL ans=0;
        for(int i=0;i<n-1;i++)
            ans=(ans+C(n,k)-C(c[i],k)-C(n-c[i],k))%mod;
        printf("%I64d\n",(ans+mod)%mod);
    }
    return 0;
}