[JSOI2008]最小生成树计数

[JSOI2008]最小生成树计数

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

题解：

首先证明一个点：两个最小生成树种长度为n的点数量是不变的。

证明很简单：如果有3个点，有两种不同的生成树，设两条边分别为a1,b1和a2,b2则有a1+b1=a2+b2

则min(a1,a2)+min(b1,b2)<=a1+b1由于是最小生成树，小于的情况不可能出现，则可证明。

那么接下来的问题就比较简单了，用最小生成树计算每种长度的边需要多少条，然后暴力计算方案数即可。

代码：

#include<bits/stdc++.h>
using namespace std;

const int max_n = 101;
const int max_m = 1001;
const int inf   = 1e9+7;
const int mod   = 31011;

struct line
{
	int l,r;
	int len;
}l[max_m];

int fa[max_n],num[max_m],cnt[max_m],size[max_n];
int n,m,a,b,c,t,limit,ans=1;

inline bool cmp(line a,line b)
{
	return a.len<b.len ? true : false;
}

int find(int a)
{
	if(a==fa[a])   return a;
	else return find(fa[a]);
}

inline void unionn(int a,int b)
{
	int a1=find(a),b1=find(b);
	if(a1==b1) return;
	if(size[a1]<size[b1]) swap(a1,b1);
	fa[b1]=a1;
	if(size[a1]==size[b1]) size[a1]++;
}

inline bool prim()
{
	int tot=0;
	for(int i=1; i<=n; ++i) fa[i]=i ,size[i]=1;
	for(int i=1; i<=m; ++i)
	{
		if(find(l[i].l)==find(l[i].r)) continue;
		unionn(l[i].l,l[i].r); 
		cnt[num[i]]++;
		if(++tot==n-1) return true;
	}
	return false;
}

int dfs(int t,int now,int last,int s)
{
	int ans=0;
	for(int i=last; i<=s; ++i)
	{
		int a1=find(l[i].l),b1=find(l[i].r);
		if(a1==b1) continue;
		unionn(a1,b1);
		if(t==now) ans+=1;
		else ans+=dfs(t+1,now,i+1,s);
		fa[a1]=a1; fa[b1]=b1;
	}
	return ans;
}

int main()
{
	freopen("bzoj_1016.in","r",stdin);
	freopen("bzoj_1016.out","w",stdout);
	scanf("%d%d",&n,&m);
	for(int i=1; i<=m; ++i)
	{
		scanf("%d%d%d",&a,&b,&c);
		l[i].l=a; l[i].r=b;
		l[i].len=c;
	}
	sort(l+1,l+m+1,cmp);
	for(int i=1; i<=m; ++i)
	{
		if(l[i].len!=l[i-1].len) t++;
		num[i]=t;
 	}
	if(!prim())
	{
		printf("0\n");
		return 0;
	}
	for(int i=1; i<=n; ++i) fa[i]=i,size[i]=1;
	int last=1;
	for(int i=1; i<=m; ++i)
	  if(l[i].len!=l[i+1].len)
	  {
	  	if(cnt[num[i]]) ans=(ans*dfs(1,cnt[num[i]],last,i))%mod;
	  	for(int j=last; j<=i; ++j) unionn(l[j].l,l[j].r);
	  	last=i+1;
	  }
	printf("%d\n",ans);
	return 0;
}