[51Nod 1677 Treecnt] 树上dfs+排列组合

[51Nod 1677 Treecnt] 树上dfs+排列组合

知识点：tree math

1. 题目链接

[51Nod 1677 Treecnt]

2. 题意描述

给定一棵n个节点的树，从1到n标号。选择k个点，你需要选择一些边使得这k个点通过选择的边联通，目标是使得选择的边数最少。
现需要计算对于所有选择k个点的情况最小选择边数的总和为多少。
样例解释：
一共有三种可能:(下列配图蓝色点表示选择的点，红色边表示最优方案中的边)
选择点{1,2}:至少要选择第一条边使得1和2联通。

选择点{1,3}:至少要选择第二条边使得1和3联通。

选择点{2,3}:两条边都要选择才能使2和3联通。

节点数目$2\le n\le 100000$.
答案对$1,000,000,007$取模。

3. 解题思路

首先，对树dfs一遍求出每个节点为根节点的子树的节点个数。用$sz[u]$表示编号为$u$的节点为根节点的子树的节点个数。那么$n - sz[u]$则表示不在该子树下的节点个数。那么$sz[u]$与$n - sz[u]$就可以表示节点$u$到节点$u$的父亲节点的边上的两端节点数。对每条边求对答案的贡献值。
然后贡献值就是从这条边的两端至少都取$1$个节点【每边至少一个顶点就保证经过这条边】，共取$k$个节点的方案数嘛。即：

$$贡献值=C_n^k-C_{sz[u]}^k - C_{n-sz[u]}^{k}$$
至此，题目已经转化为排列组合的问题了。观察式子发现， $k$固定。可以$O(n*log(MOD))$预处理出 $C_x^k$。
当然，这题通过对逆元预处理，可以达到 $O(n)+O(MOD)$。请参见：《 51Nod 逆推求阶乘逆元》

4. 实现代码

#include <bits/stdc++.h>
using namespace std;

typedef long long LL;
typedef long double LB;
typedef pair<int, int> PII;
typedef pair<LL, LL> PLL;

const int MAXN = 100000 + 5;
const int INF = 0x3f3f3f3f;
const LL MOD = 1000000007;
const LL INFL = 0x3f3f3f3f3f3f3f3fLL;
const LB eps = 1e-8;

LL quick_pow(LL a, LL b, LL mod) {
    LL ret = 1;
    while(b > 0) {
        if(b & 1) ret = ret * a % mod;
        a = a * a % mod;
        b >>= 1;
    }
    return ret;
}
LL inv(LL a, LL mod) { return quick_pow(a, mod - 2, mod); }

struct Edge {
    int v, next;
} edge[MAXN << 1];
int head[MAXN], sz[MAXN], tot;
int n, k;
LL F[MAXN]; /// F[n] 表示C(n, k);

void init() {
    tot = 0;
    memset(head, -1, sizeof(head));
    memset(F, 0, sizeof(F));
    LL dw = 1, up = 1;
    for(int i = 1; i <= k; ++i) {
        dw = dw * inv(i, MOD) % MOD;
        up = up * i % MOD;
    }
    F[k] = 1;
    for(int i = k + 1; i <= n; ++i) {
        up = up * i % MOD * inv(i - k, MOD) % MOD;
        F[i] = up * dw % MOD;
    }
}

inline void add_edge(int& u, int& v) {
    edge[tot] = Edge{v, head[u]};
    head[u] = tot ++;
}
void dfs(int u, int fa) {
    sz[u] = 1;
    for(int i = head[u], v; ~i; i = edge[i].next) {
        v = edge[i].v;
        if(v == fa) continue;
        dfs(v, u);
        sz[u] += sz[v];
    }
}

int main() {
#ifdef ___LOCAL_WONZY___
    freopen("input.txt", "r", stdin);
#endif // ___LOCAL_WONZY___
    int u, v;
    while(~scanf("%d %d", &n, &k)) {
        init();
        for(int i = 1; i < n; ++i) {
            scanf("%d %d", &u, &v);
            add_edge(u, v), add_edge(v, u);
        }
        if(k == 1) {
            printf("0\n");
            continue;
        }
        dfs(1, -1);
        LL ans = 0;
        LL z = F[n]; /// C(n, k);
        for(int i = 2; i <= n; ++i) {
            if(sz[i] < 1 || n - sz[i] < 1) continue;
            LL a = F[sz[i]];
            LL b = F[n - sz[i]];
            ans += (z - a - b + MOD) % MOD;
            ans %= MOD;
        }
        printf("%lld\n", ans);
    }
    return 0;
}