1. 程式人生 > >BZOJ 1016--最小生成樹計數(深搜&kruskal)

BZOJ 1016--最小生成樹計數(深搜&kruskal)

names 連通性 如果 int 沒有 計數 ++ struct include

    想我這樣的zz根本不會矩陣樹。。。。。

題目鏈接:

    http://www.lydsy.com/JudgeOnline/problem.php?id=1016

Solution

    首先,如果用上所有邊都做不出最小生成樹,答案顯然是0。。

    然後,對於一個圖的所有最小生成樹,他們有的權值為w的邊數一定是一樣多的。。

    所以先求出一個最小生成樹,然後把其中一條邊去掉,再搜索一下在相同權值的邊中是否仍然可以保持連通性。

    根據乘法原理求出答案。。。

    然後就沒有然後了。。。

代碼

#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#include<iostream>
#define M 1010
#define mod 31011
using namespace std;
int n,m,cnt,tot;
int f[M],w[M],sum[M];
bool c[M];
struct edge{
    int l,r,w;
}e[M],a[M];
bool cmp(edge p,edge q){return p.w<q.w;}
int find(int x){
    if(f[x]==x) return f[x];
    return find(f[x]);
}
void RE(){for(int i=1;i<=n;i++) f[i]=i;}
void dfs(int x,int now,int k){
    if(now==a[x].r+1){
        if(k==a[x].w)tot++;
        return;
    }
    int xx=find(e[now].l),yy=find(e[now].r);
    if(xx!=yy){
        f[xx]=yy;
        dfs(x,now+1,k+1);
        f[xx]=xx;f[yy]=yy;
    }
    dfs(x,now+1,k);
}
int main(){
    int ans=1;
    cnt=0;tot=0;
    int xx,yy;
    scanf("%d%d",&n,&m);
    for(int i=1;i<=m;i++)
        scanf("%d%d%d",&e[i].l,&e[i].r,&e[i].w);
    sort(e+1,e+1+m,cmp);
    RE();
    for(int i=1;i<=m;i++){
        if(e[i].w!=e[i-1].w){a[cnt].r=i-1;cnt++;a[cnt].l=i;}
        xx=find(e[i].l);yy=find(e[i].r);
        if(xx!=yy){f[xx]=yy;a[cnt].w++;tot++;}
    }
    a[cnt].r=m;
    if(tot+1!=n){printf("0\n");return 0;}
    RE();
    for(int i=1;i<=cnt;i++){
        tot=0;
        dfs(i,a[i].l,0);
        ans=(ans*tot)%mod;
        for(int j=a[i].l;j<=a[i].r;j++){
            xx=find(e[j].l);yy=find(e[j].r);
            if(xx!=yy) f[xx]=yy;
        }
    }
    printf("%d\n",ans);
    return 0;
}

  

  

This passage is made by Iscream-2001.

BZOJ 1016--最小生成樹計數(深搜&kruskal)