BZOJ1016: [JSOI2008]最小生成树计数 Kruskal

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

Time Limit: 1 Sec   Memory Limit: 162 MB

Submit: 4815   Solved: 1958

[Submit][Status][Discuss]

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,0
00。数据保证不会出现自回边和重边。注意：具有相同权值的边不会超过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

题解：

我们可以发现，当存在多个最小生成树时，它们每条边的权值和权值相同的边的数量都是一定的，所以我们可以把权值相同的边放在一起处理。

我们先按权值的大小排序，对于权值相同的边，一定要选出其中一些并使连通块的数量尽量的减少，所以我们暴力枚举所有边选或者不选最多是2^10，记录下来使连通块减少到最少的所有最优方案，而且对于这些最优方案它们对图的连通性影响是完全相同的（这个可以自己证一下），然后也要注意并查集不能路径压缩，因为对于每一种情况我们操作完之后要给它复原，具体细节就不多说了。

#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int MOD=31011;
const int N=1005;
const int inf=1707185547;
struct Tree{int f,t,w;}e[N];
int n,m,cnt,fa[N],tfa[N],f[N],tot,ct,first,final,mx,now,haha;
bool b[N],da[N];
int find(int x)
{
	if(fa[x]==x) return x;
	return find(fa[x]);
}
void insert(int f,int t,int w)
{
	cnt++;
	e[cnt].f=f;
	e[cnt].t=t;
	e[cnt].w=w;
}
void dfs(int deep)
{
	if(deep>final)
	{
		tot=0,ct=0;
		for(int i=first;i<=final;i++)
		if(b[i])
		{
			int fx=find(e[i].f),fy=find(e[i].t);
			if(fx!=fy) tot++,fa[fy]=fx;
			ct++;
		}
		for(int i=1;i<=n;i++) fa[i]=tfa[i];
		if(ct!=tot) return;
		if(tot>mx)
		{
		    mx=tot,now=1;
		    for(int i=first;i<=final;i++) da[i]=b[i];
		}
		else if(tot==mx) now++;
		return;
	}
	for(int i=0;i<=1;i++)
	{
		b[deep]=i;
		dfs(deep+1);
	}
}
bool cmp(Tree u,Tree v)
{return u.w<v.w;}
int main()
{
	scanf("%d%d",&n,&m);
	int x,y,z;
	for(int i=1;i<=m;i++)
	{
		scanf("%d%d%d",&x,&y,&z);
		insert(x,y,z);
	}
	sort(e+1,e+m+1,cmp);
	for(int i=1;i<=n;i++) fa[i]=i;
	int ans=1,total=n;
	for(int i=1;i<=m;i++)
	{
		first=i;
		now=0,mx=0;
		while(e[i].w==e[i+1].w) i++;
		final=i;
		for(int j=1;j<=n;j++) tfa[j]=fa[j];
		dfs(first);
		for(int j=first;j<=final;j++)
		if(da[j])
		{
			int fx=find(e[j].f),fy=find(e[j].t);
			if(fx!=fy) fa[fy]=fx,total--;
		}
	    ans=ans*now%MOD;
	}
	if(total!=1) printf("0");
	else printf("%d",ans);
}