bzoj1016 [JSOI2008]最小生成树计数

Description

现在给出了一个简单无向加权图。你不满足于求出这个图的最小生成树，而希望知道这个图中有多少个不同的最小生成树。（如果两颗最小生成树中至少有一条边不同，则这两个最小生成树就是不同的）。由于不同的最小生成树可能很多，所以你只需要输出方案数对$31011$的模就可以了。

Solution

求最小生成树的个数。先跑一遍kruskal，求出每种边权在最小生成树出现了多少次。这里有一个结论，不同的最小生成树所用的边权是一一对应的。所以跑完kruskal后在dfs就好了。

#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;

inline int read() {
    int x = 0, flag = 1; char ch = getchar();
    while (ch > '9' || ch < '0') { if (ch == '-') flag = -1; ch = getchar(); }
    while (ch <= '9' && ch >= '0') { x = x * 10 + ch - '0'; ch = getchar(); }
    return x * flag;
}

#define N 1005
#define rep(ii, aa, bb) for (int ii = aa; ii <= bb; ii++)
#define ll long long
const ll ha = 31011;

int n, m;
int fa[N];
struct edgeType { int u, v, w; } e[N];
bool cmp(const edgeType x, const edgeType y) { return x.w < y.w; }
struct saveType { int l, r, w; } s[N];
int tot;
int sum;

int find(int x) { return x == fa[x] ? x : find(fa[x]); }

void dfs(int x, int pos, int cnt) {
    if (pos == s[x].r + 1) {
        sum += (cnt == s[x].w);
        return;
    }
    int u = find(e[pos].u), v = find(e[pos].v);
    if (u != v) {
        fa[u] = v;
        dfs(x, pos + 1, cnt + 1);
        fa[u] = u; fa[v] = v;
    }
    dfs(x, pos + 1, cnt);
}

int main() {
    cin >> n >> m;
    rep(i, 1, n) fa[i] = i;
    rep(i, 1, m) e[i].u = read(), e[i].v = read(), e[i].w = read();
    sort(e + 1, e + 1 + m, cmp);
    int cnt = 0;
    rep(i, 1, m) {
        if (e[i].w != e[i - 1].w) { s[++tot].l = i; s[tot - 1].r = i - 1; }
        int u = find(e[i].u), v = find(e[i].v);
        if (u != v) { cnt++; s[tot].w++; fa[u] = v; }
    }
    s[tot].r = m;
    if (cnt != n - 1) { putchar('0'); return 0; }
    rep(i, 1, n) fa[i] = i;
    int ans = 1;
    rep(i, 1, tot) {
        sum = 0;
        dfs(i, s[i].l, 0);
        (ans *= sum) %= ha;
        rep(j, s[i].l, s[i].r) {
            int u = find(e[j].u), v = find(e[j].v);
            if (u != v) fa[u] = v;
        }
    }
    cout << ans;
    return 0;
}