bzoj1016: [JSOI2008]最小生成树计数

题目

　　http://www.lydsy.com/JudgeOnline/problem.php?id=1016

题解

　　求最小生成树的个数，因为这个题有个很好的条件“具有相同权值的边不会超过10条。”所以可以爆搜。

　　有个定理是说，所有最小生成树中同种边权的边的使用次数相同且作用相同。举个例子，比如用kruskal求出边权为2的有3条边，且这三条边连通了集合A、B、C、D，那么其他的最小生成树中，边权为2的边永远只有3条且都连通集合A、B、C、D。

　　根据上述原理，就可以把边按边权进行分类，爆搜出所有合法的方案，然后乘法原理乘起来就是答案。

代码

//生成树
#include <cstdio>
#include <algorithm>
#include <cstdlib>
#define maxm 1010
#define p 31011
using namespace std;
int N, M, f[maxm], ans, sum, xian;
struct edge{int a, b, v, cnt;}e[maxm];
int find(int x){return x==f[x]?x:find(f[x]);}
bool operator<(edge a, edge b){return a.v<b.v;}
void input()
{
	int i;
	scanf("%d%d",&N,&M);
	for(i=1;i<=M;i++)scanf("%d%d%d",&e[i].a,&e[i].b,&e[i].v);
}
void kruskal()
{
	int i, j, fa, fb, x, t=0;
	sort(e+1,e+M+1);
	for(i=1;i<=N;i++)f[i]=i;
	for(i=1;i<=M;i++)
	{
		fa=find(e[i].a),fb=find(e[i].b);
		if(fa!=fb)
		{
			f[fa]=fb;
			e[i].cnt=1;
			t++;
		}
		if(e[i].v!=e[i+1].v)
		{
			xian=i;
			x=0;
			for(j=i;e[j].v==e[i].v;j--)x+=e[j].cnt;
			for(j=i;e[j].v==e[i].v;j--)e[j].cnt=x;
			if(t==N-1)break;
		}
	}
	if(t!=N-1){printf("0\n");exit(0);}
	for(i=1;i<=N;i++)f[i]=i;
}
void dfs(int r, int now, int cnt)
{
	if(now>r)
	{
		if(cnt==e[r].cnt)sum++;
		return;
	}
	int fa=find(e[now].a), fb=find(e[now].b);
	if(fa!=fb)
	{
		f[fa]=fb;
		dfs(r,now+1,cnt+1);
		f[fa]=fa;
	}
	dfs(r,now+1,cnt);
}
void work()
{
	int i, j, fa, fb;
	ans=1;
	for(i=1;i<=xian;i=j)
	{
		for(j=i;e[j].v==e[i].v;j++);
		sum=0;
		dfs(j-1,i,0);
		ans=(ans*sum)%p;
		for(j=i;e[j].v==e[i].v;j++)
		{
			fa=find(e[j].a),fb=find(e[j].b);
			if(fa!=fb)f[fa]=fb;
		}
	}
}
int main()
{
	input();
	kruskal();
	work();
	printf("%d\n",ans);
	return 0;
}