【bzoj1016】【JSOI2008】【最小生成树计数】【dfs+最小生成树】

Description

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

Input

第一行包含两个数，n和m，其中1<=n<=100; 1<=m<=1000; 表示该无向图的节点数和边数。每个节点用1~n的整数编号。接下来的m行，每行包含两个整数：a, b, c，表示节点a, b之间的边的权值为c，其中1<=c<=1,000,000,000。数据保证不会出现自回边和重边。注意：具有相同权值的边不会超过10条。

Output

输出不同的最小生成树有多少个。你只需要输出数量对31011的模就可以了。

Sample Input

4 6
1 2 1
1 3 1
1 4 1
2 3 2
2 4 1
3 4 1

Sample Output

8

题解：首先一种权值的边在最小生成树中的作用是一样的。

即控制的点的数量是一样的。

所以我们先做一遍最小生成树。统计一下每中权值的边有多少条，控制几个点。

然后对于每种权值的边。我们dfs一下看看有多少种等价的方案。

然后用乘起来即可。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define P 31011
using namespace std;
int n,m,cnt,tt,ans(1),temp,fa[505];
struct use{int st,en,v;}e[10005];
struct use2{int l,r,t;}s[10005];
bool cmp(use a,use b){return a.v<b.v;}
int find(int x){return x==fa[x]?x:find(fa[x]);}
void dfs(int k,int x,int pp){
     if(x==s[k].r+1){if(pp==s[k].t)temp++;return;}
     int p=find(e[x].st),q=find(e[x].en);
     if(p!=q){fa[p]=q;dfs(k,x+1,pp+1);fa[p]=p;fa[q]=q;}
     dfs(k,x+1,pp);
}
int main(){
    scanf("%d%d",&n,&m);
    for (int i=1;i<=m;i++){scanf("%d%d%d",&e[i].st,&e[i].en,&e[i].v);}
    for(int i=1;i<=n;i++)fa[i]=i;sort(e+1,e+m+1,cmp);
    for(int i=1;i<=m;i++){
        if(e[i].v!=e[i-1].v){s[++cnt].l=i;s[cnt-1].r=i-1;}
        int p=find(e[i].st),q=find(e[i].en);
        if(p!=q){fa[p]=q;s[cnt].t++;tt++;}
    }
    s[cnt].r=m;if(tt!=n-1){printf("0");return 0;}
    for(int i=1;i<=n;i++)fa[i]=i;
    for(int i=1;i<=cnt;i++){
        temp=0;dfs(i,s[i].l,0);(ans*=temp)%=P;//cout<<sum<<endl;
        for(int j=s[i].l;j<=s[i].r;j++){
            int p=find(e[j].st),q=find(e[j].en);
            if(p!=q)fa[p]=q;
        }
    }
    printf("%d",ans);
}