【kruscal】【最小生成树】【搜索】bzoj1016 [JSOI2008]最小生成树计数

不用Matrix-tree定理什么的，一边kruscal一边 对权值相同的边 暴搜即可。将所有方案乘起来。

 #include<cstdio>
 #include<algorithm>
 using namespace std;
 int n,m;
 struct Disjoint_Set
 {
     int fa[],rank[];
     void init(){for(int i=;i<=n;i++) fa[i]=i;}
     int findroot(int x)
       {
         if(fa[x]==x) return x;
         int rt=findroot(fa[x]);
         fa[x]=rt;
         return rt;
       }
     void Union(int U,int V)
       {
         if(rank[U]<rank[V]) fa[U]=V;
         else
           {
             fa[V]=U;
             if(rank[U]==rank[V]) rank[U]++;
           }
       }
 };
 Disjoint_Set S,used;
 struct Edge{int u,v,w;};
 bool cmp(const Edge &a,const Edge &b){return a.w<b.w;}
 Edge edges[];
 int res,ans=,tot,cnt,sta,end;
 void dfs(int cur,int sum,Disjoint_Set now)
 {
     if(cur>end)
       {
           if(sum==cnt) res++;
           return;
       }
     dfs(cur+,sum,now);
     int f1=now.findroot(edges[cur].u),f2=now.findroot(edges[cur].v);
     if(f1!=f2) {now.Union(f1,f2); dfs(cur+,sum+,now);}
 }
 int main()
 {
     scanf("%d%d",&n,&m);
     for(int i=;i<=m;i++) scanf("%d%d%d",&edges[i].u,&edges[i].v,&edges[i].w);
     sort(edges+,edges+m+,cmp);
     S.init();used.init();
     for(int i=;i<=m;i++)
       {
           if(edges[i].w!=edges[i-].w) {used=S; cnt=; sta=i;}
           int f1=S.findroot(edges[i].u),f2=S.findroot(edges[i].v);
           if(f1!=f2) {S.Union(f1,f2); tot++; cnt++;}
           if(edges[i].w!=edges[i+].w)
             {
                 res=; end=i;
                 dfs(sta,,used);
                 ans=((ans%)*(res%))%;
             }
           else if(tot==n-)
             {
                 res=;
                 for(int j=i+;j<=m;j++)
                   if(edges[j].w!=edges[i].w)
                     {
                       end=j-;
                       goto OUT;
                     }
                 end=m;
                 OUT:dfs(sta,,used);
                 ans=((ans%)*(res%))%;
                 break;
             }
       }
     printf("%d\n",tot==n- ? ans : );
     return ;
 }