51nod - 1677 treecnt

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