BZOJ1016(JSOI2008)[最小生成树计数]--kruskal+Dfs+并查集

【链接】
bzoj1016

【解题报告】

对于这题，应该知道一个无向图中的所有最小生成树的每种权值的边数量相同。

那么我们只需要dfs一遍算出每种边的方案书，再相乘就行了。

Ps:dfs中的并查集不能路压，因为要回退QAQ

#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn=105,maxm=1005,tt=31011;
int n,m,sum,ans,cnt,tot,fa[maxn];
struct wjd
{
    int x,y,z;
    bool operator < (const wjd &a) const{
        return z<a.z;
    }
}a[maxm];
struct ZigZagK
{
    int l,r,v;
}c[maxm];
inline char nc()
{
    static char buf[100000],*l,*r;
    if (l==r) r=(l=buf)+fread(buf,1,100000,stdin);
    if (l==r) return EOF; return *l++;
}
inline int Read()
{
    int res=0,f=1; char ch=nc(),cc=ch;
    while (ch<'0'||ch>'9') cc=ch,ch=nc();
    if (cc=='-') f=-1;
    while (ch>='0'&&ch<='9') res=res*10+ch-48,ch=nc();
    return res*f;
}
int Getfa(int x)
{
    if (x==fa[x]) return x;
    return Getfa(fa[x]);
}
void Dfs(int x,int now,int k)
{
    if (now>c[x].r) {if (k==c[x].v) sum++; return;}
    int fx=Getfa(a[now].x),fy=Getfa(a[now].y);
    if (fx!=fy) fa[fx]=fy,Dfs(x,now+1,k+1),fa[fx]=fx,fa[fy]=fy;
    Dfs(x,now+1,k);
}
int main()
{
    freopen("1016.in","r",stdin);
    freopen("1016.out","w",stdout);
    n=Read(); m=Read();
    for (int i=1; i<=m; i++) a[i]=(wjd){Read(),Read(),Read()};
    sort(a+1,a+1+m);
    for (int i=1; i<=n; i++) fa[i]=i;
    cnt=tot=0; ans=1;
    for (int i=1; i<=m; i++)
    {
        if (a[i].z!=a[i-1].z) c[cnt].r=i-1,c[++cnt].l=i;
        int x=Getfa(a[i].x),y=Getfa(a[i].y);
        if (x!=y) fa[x]=y,c[cnt].v++,tot++;
    }
    c[cnt].r=m;
    if (tot!=n-1) {printf("0"); return 0;}
    for (int i=1; i<=n; i++) fa[i]=i;
    for (int i=1; i<=cnt; i++)
    {
        if (!c[i].v) continue;
        sum=0; Dfs(i,c[i].l,0); ans=(ans*sum)%tt;
        for (int j=c[i].l,x,y; j<=c[i].r; j++)
         x=Getfa(a[j].x),y=Getfa(a[j].y),fa[x]=y;
    }
    printf("%d",ans);
    return 0;
}