51NOD 1677 treecnt

1677 treecnt
基准时间限制：1 秒 空间限制：131072 KB 分值: 40 难度：4级算法题 收藏  关注
给定一棵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

对任意边(u,v)
设a=以v为根的子树的点
b=n-a
那这条边被选择的次数=C(a,1)*C(b,k-1)+C(a,2)*C(b,k-2)+C(a,3)*C(b,k-3)+…..
显然 这样肯定会TLE
不妨换个角度
考虑从n个点中选择k个点 一共有C(n,k)总情况
当k个点全在a中选出来 或 k个点全在b中选出来的情况是要排除的
所以这条边被选择的次数为C(n,k)-C(a,k)-C(b,k)

#include<iostream>
#include<stdlib.h>
#include<stdio.h>
#include<string>
#include<vector>
#include<deque>
#include<queue>
#include<algorithm>
#include<set>
#include<map>
#include<stack>
#include<time.h>
#include<math.h>
#include<list>
#include<cstring>
#include<fstream>
//#include<memory.h>
using namespace std;
#define ll long long
#define ull unsigned long long
#define pii pair<int,int>
#define INF 1000000007
#define pll pair<ll,ll>
#define pid pair<int,double>

const int N=100000+5;

int head[N];
struct Edge{
    int to,next;
    int num;
}edge[2*N];

inline void addEdge(int k,int u,int v){
    edge[k].to=v;
    edge[k].next=head[u];
    head[u]=k;
}

ll egcd(ll a,ll b,ll&x,ll&y){
    if(b==0){
        x=1,y=0;
        return a;
    }
    else{
        ll d=egcd(b,a%b,x,y);
        ll xt=x,yt=y;
        x=yt;
        y=xt-a/b*yt;
        return d;
    }
}

ll fac[N];

inline ll C(int n,int k){
    if(n==k)
        return 1;
    if(n<k)
        return 0;
    ll x,y;
    ll fz=fac[n];
    ll fm=(fac[k]*fac[n-k])%INF;
    egcd(fm,INF,x,y);
    x=(x+INF)%INF;
    ll ans=(fz*x)%INF;
    return ans;
}

void init(int n){
    fill(head,head+n+1,-1);
    fac[0]=1;
    for(int i=1;i<=n;++i){
        fac[i]=(fac[i-1]*i)%INF;
    }
}

inline int dfs(int u,int father){
    int sum=1;
    for(int i=head[u];i!=-1;i=edge[i].next){
        int v=edge[i].to;
        if(v==father)
            continue;
        int num=dfs(v,u);
        edge[i].num=edge[i^1].num=num;
        sum+=num;
    }
    return sum;
}

ll slove(int n,int k){
    ll sum=C(n,k);
    dfs(1,-1);
    ll ans=0;
    for(int i=0;i<2*(n-1);i+=2){
        int a=edge[i].num;
        int b=n-a;
        ll tmp=sum-C(a,k)-C(b,k);
        ans=(ans+tmp)%INF;
    }
    return (ans+INF)%INF;
}

int main()
{
    //freopen("/home/lu/文档/r.txt","r",stdin);
    //freopen("/home/lu/文档/w.txt","w",stdout);
    int n,k;
    scanf("%d%d",&n,&k);
    init(n);
    for(int i=0,u,v;i<n-1;++i){
        scanf("%d%d",&u,&v);
        addEdge(2*i,u,v);
        addEdge(2*i+1,v,u);
    }
    printf("%lld\n",slove(n,k));
    return 0;
}