MST——BZOJ1016 [JSOI2008]最小生成树计数

http://www.lydsy.com/JudgeOnline/problem.php?id=1016
我们来分析一下这个最小生成树
如果两颗最小生成树中至少有一条边不同，则这两个最小生成树就是不同的
因为说相同边权的边的编号不影响最小生成树的大小，这个我们就可以算进多个答案了
所以我们可以先把边排序，跑一遍Kruskal记录一下相同边权的边对答案产生影响的数目还有起始终止位置（排序后）
然后我们每种边权跑一个dfs，O（n）跑一遍暴力计算全部边能影响答案的数目
最后乘起来就好了
需要注意到的是，并查集不能状态压缩，否则dfs回溯时会产生错误

#include<bits/stdc++.h>
using namespace std;
const int MOD=31011;
struct edge{
    int x,y,v;
}a[1001];
inline bool cmp(edge a,edge b){return a.v<b.v;}
int n,m,fa[101],sum=0,cnt=0,l[1001]={0},r[1001]={0},v[1001]={0};
inline int getfather(int v){
    if(fa[v]==v)return v;
    return getfather(fa[v]);
}
inline void dfs(int i,int x,int s){
    if(x==r[i]+1){
        if(s==v[i])cnt++;
        return;
    }
    int xx=getfather(a[x].x),yy=getfather(a[x].y);
    if(xx!=yy){fa[xx]=yy;dfs(i,x+1,s+1);fa[xx]=xx;}
    dfs(i,x+1,s);
}
int ans=1;
int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)scanf("%d%d%d",&a[i].x,&a[i].y,&a[i].v);
    sort(a+1,a+m+1,cmp);
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=m;i++){
        if(a[i].v!=a[i-1].v)r[sum]=i-1,l[++sum]=i;
        int xx=getfather(a[i].x),yy=getfather(a[i].y);
        if(xx!=yy)fa[xx]=yy,cnt++,v[sum]++;
    }
    r[sum]=m;
    if(cnt<n-1){
        printf("0");
        return 0;
    }
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=sum;i++){
        cnt=0;
        dfs(i,l[i],0);
        (ans*=cnt)%=MOD;
        for(int j=l[i];j<=r[i];j++){
            int xx=getfather(a[j].x),yy=getfather(a[j].y);
            if(xx!=yy)fa[xx]=yy;
        }
    }
    printf("%d",ans);
    return 0;
}