[FROM WOJ]BZOJ1016[JSOI2008]最小生成树计数

BZOJ1016[JSOI2008]最小生成树计数

n，m数据规模是骗人的，我信了然后RE了一个点，所以没写

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

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

样例输入
4 6
1 2 1
1 3 1
1 4 1
2 3 2
2 4 1
3 4 1

样例输出
8

SOL
对于一个图的最小生成树，它的每种权值的边使用的数量一定，树的形态也是一定的，于是先跑一遍Kruskal，再DFS统计答案。
完了。

代码：

#include<bits/stdc++.h>
#define mod 31011
#define N 10001
using namespace std;
int n,m,tot,ans=1,sum,cnt;
int f[N];
struct line{
	int u,v,w;
}e[N];
struct line_value{
	int st,ed,num;
}a[N];
inline bool cmp(line a,line b){return a.w<b.w;}
int find(int x){return f[x]==x ? x : find(f[x]);}
void dfs(int x,int now,int t){
	if(now==a[x].ed+1){
		if(t==a[x].num)sum++;
		return;
	}
	int fu=find(e[now].u),fv=find(e[now].v);
	if(fu!=fv){
		f[fu]=fv;
		dfs(x,now+1,t+1);
		f[fu]=fu;f[fv]=fv;
	}
	dfs(x,now+1,t);
	return;
}
int main(){
	scanf("%d%d",&n,&m);
	for(int register i=1;i<=n;i++)f[i]=i;
	for(int register i=1;i<=m;i++)scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
	sort(e+1,e+m+1,cmp);
	for(int register i=1;i<=m;i++){
		if(e[i].w!=e[i-1].w){
			a[cnt].ed=i-1;
			a[++cnt].st=i;
		}
		int fx=find(e[i].u),fy=find(e[i].v);
		if(fx!=fy){
			f[fx]=fy;
			a[cnt].num++;
			tot++;
		}
	}
	a[cnt].ed=m;
	if(tot!=n-1){puts("0");return 0;}
	for(int register i=1;i<=n;i++)f[i]=i;
	for(int register i=1;i<=cnt;i++){
		sum=0;
		dfs(i,a[i].st,0);
		ans=(ans*sum)%mod;
		for(int register j=a[i].st;j<=a[i].ed;j++){
			int fu=find(e[j].u),fv=find(e[j].v);
			if(fu!=fv)f[fu]=fv;
		}
	}
	printf("%d",ans);
	return 0;
}