[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;
}