【bzoj1016】 JSOI2008—最小生成树计数

http://www.lydsy.com/JudgeOnline/problem.php?id=1016 (题目链接)

题意：求图的最小生成树计数。

Solution

　　%了下题解，发现要写矩阵树，150++的程序什么鬼。于是就蒯了hzwer的简便方法。

　　将边按照权值大小排序，将权值相同的边分到一组，统计下每组分别用了多少条边。然后对于每一组进行dfs，判断是否能够用这一组中的其他边达到相同的效果。最后把每一组的方案数相乘就是答案。

　　注意并查集不要压缩路径，不然的话不好回溯。

代码：

// bzoj1016
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#define LL long long
#define inf 2147483640
#define MOD 31011
#define Pi acos(-1.0)
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
using namespace std;
int getint() {
int f=1,x=0;char ch=getchar();
while (ch<='0' || ch>'9') {if (ch=='-') f=-1;ch=getchar();}
while (ch>='0' && ch<='9') {x=x*10+ch-'0';ch=getchar();}
return x*f;
}

const int maxn=1010;
struct edge {int u,v,w;}e[maxn<<2],a[maxn];
int fa[maxn],sum,n,m,cnt,tot;

bool cmp(edge a,edge b) {
return a.w<b.w;
}
int find(int x) {
return fa[x]==x ? x : find(fa[x]);
}
void dfs(int x,int now,int k) {
if (now==a[x].v+1) {
if (k==a[x].w) sum++;
return;
}
int r1=find(e[now].u),r2=find(e[now].v);
if (r1!=r2) {
fa[r1]=r2;
dfs(x,now+1,k+1);
fa[r1]=r1,fa[r2]=r2;
}
dfs(x,now+1,k);
}
int main() {
scanf("%d%d",&n,&m);
for (int i=1;i<=m;i++) scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].w);
for (int i=1;i<=n;i++) fa[i]=i;
sort(e+1,e+1+m,cmp);
for (int i=1;i<=m;i++) {
if (e[i].w!=e[i-1].w) {a[++cnt].u=i;a[cnt-1].v=i-1;}
int r1=find(e[i].u),r2=find(e[i].v);
if (r1!=r2) {fa[r1]=r2;a[cnt].w++;tot++;}
}
a[cnt].v=m;
if (tot!=n-1) {printf("0");return 0;}
for (int i=1;i<=n;i++) fa[i]=i;
int ans=1;
for (int i=1;i<=cnt;i++) {
sum=0;
dfs(i,a[i].u,0);
ans=(ans*sum)%MOD;
for (int j=a[i].u;j<=a[i].v;j++) {
int r1=find(e[j].u),r2=find(e[j].v);
if (r1!=r2) fa[r1]=r2;
}
}
printf("%d",ans);
return 0;
}