首页 > 代码库 > 【bzoj1016】 JSOI2008—最小生成树计数

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

  

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