BZOJ 1016 最小生成树计数 暴力

题意:给定一张简单无向加权图，求其最小生成树方案数.n<=100,m<=1000.边权相同的<=10.

结论:MST不同,但包含边值都是一样的.比如MST有3条边权为2的边.其他任意MST也都正好有3条边权为2的边.

因为添加完某种val边 形成的连通分量个数是一定的.先求出某个MST,记录val边使用次数,然后对每种val边暴力枚举有多少种方案.累乘即可.

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
typedef pair<int,int> ii;
const int N=3e4+20,inf=0x3f3f3f3f,mod=31011;
struct edge{
	int a,b,c;
	bool operator < (const edge &b)const{return c<b.c;}
}e[N],t[N];
int n,m,cnt,tot,ans=1,f[N];
int find(int x){return x==f[x]?x:find(f[x]);}
int dfs(int x,int p,int cnt)
{
	if(p==t[x].b+1)	return cnt==t[x].c;
	int sum=0;
	int a=find(e[p].a),b=find(e[p].b);
	if(a!=b)
	{
		f[a]=b;
		sum+=dfs(x,p+1,cnt+1);
		f[a]=a,f[b]=b;//	
	} 
	return sum+dfs(x,p+1,cnt);
}
int main()
{
	cin>>n>>m;
	for(int i=1;i<=n;i++) f[i]=i;
	for(int i=1;i<=m;i++)	scanf("%d%d%d",&e[i].a,&e[i].b,&e[i].c);
	sort(e+1,e+1+m);
	for(int i=1;i<=m;i++)
	{
		int a=find(e[i].a),b=find(e[i].b);
		if(e[i].c!=e[i-1].c)
		{
			t[cnt].b=i-1;
			t[++cnt].a=i;
		}
		if(a!=b)
		{
			f[a]=b;
			t[cnt].c++,tot++;
		}
	}
	if(tot!=n-1)
	{
		puts("0");
		return 0;
	}
	for(int i=1;i<=n;i++)	f[i]=i;
	t[cnt].b=m;
	for(int i=1;i<=cnt;i++)
	{
		ans=(ans*dfs(i,t[i].a,0))%mod;
		for(int j=t[i].a;j<=t[i].b;j++)
		{
			int a=find(e[j].a),b=find(e[j].b);
			if(a!=b)
				f[a]=b;
		}
	}
	cout<<ans<<endl;

	
	return 0;
}

点击打开链接

法2 val边形成联通分量 缩点后 用Matrix-Tree定理技术.