bzoj1006 最小生成树计数

分析：我们可以先做一遍kruskal，得出每种权值使用的数量，然后用权值相同的边去替代，用乘法原理去算就可以了。

#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(int i=a;i<=b;i++)
#define fd(i,a,b) for(int i=a;i>=b;i--)
using namespace std;
int n,m;
const int mo=31011;
const int N=1e5+5;
struct node
{
    int x,y,v;
}e[N*4];
struct data
{
    int l,r,v;
}a[N];
int fa[N];
int cnt,tot,ans=1,sum;
bool cmp(node x,node y)
{
    return x.v<y.v;
}
int find(int x)
{
    return x==fa[x]?x:find(fa[x]);
}
inline void dfs(int x,int now,int k)
{
    if (now==a[x].r+1)
    {
        if(k==a[x].v)sum++;
        return ;
    }
    int p=find(e[now].x),q=find(e[now].y);
    if (p!=q)
    {
        fa[p]=q;
        dfs(x,now+1,k+1);
        fa[p]=p;
        fa[q]=q;
    }
    dfs(x,now+1,k);
}
int main()
{
    scanf("%d%d",&n,&m);
    fo(i,1,n)fa[i]=i;
    fo(i,1,m)scanf("%d%d%d",&e[i].x,&e[i].y,&e[i].v);
    sort(e+1,e+m+1,cmp);
    fo(i,1,m)
    {
        if (e[i].v!=e[i-1].v)
        {
            a[++cnt].l=i;
            a[cnt-1].r=i-1;
        }
        int p=find(e[i].x),q=find(e[i].y);
        if (p!=q)
        {
            fa[p]=q;
            a[cnt].v++;
            tot++;
        }
    }
    a[cnt].r=m;
    if (tot!=n-1)
    {
        printf("0");
        return 0;
    }
    fo(i,1,n)
    {
        fa[i]=i;
    }
    fo(i,1,cnt)
    {
        sum=0;
        dfs(i,a[i].l,0);
        ans=(ans*sum)%mo;
        fo(j,a[i].l,a[i].r)
        {
            int p=find(e[j].x),q=find(e[j].y);
            if (p!=q)fa[p]=q;
        }
    }
    printf("%d",ans);
}