题意:给定一张简单无向加权图,求其最小生成树方案数.n<=100,m<=1000.边权相同的<=10.
因为添加完某种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定理技术.