HDU 4408 Minimum Spanning Tree（最小生成树计数）

题目链接：http://acm.hdu.edu.cn/showproblem.php?pid=4408

题意：求最小生成树个数。

思路：模板

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <algorithm>
 #include <vector>
 #define int64 long long
 using namespace std;
 
 
 const int MAX=105;    //点的个数
 const int MAXE=1005;  //边的个数
 struct node
 {
     int set[MAX];
 
     void init(int n)
     {
         int i;
         for(i=0;i<=n;i++) set[i]=i;
     }
 
     int find(int x)
     {
         if(set[x]!=x) set[x]=find(set[x]);
         return set[x];
     }
 
     int Union(int x,int y)
     {
         int xx=find(x);
         int yy=find(y);
         if(xx==yy) return -1;
         set[xx]=yy;
         return 1;
     }
 };
 
 
 struct Node
 {
     int u,v,dis;
 };
 
 node a,b,c;
 int n,m;
 Node e[MAXE];
 int visit[MAX];
 vector<int> g[MAX];
 int64 p[MAX][MAX],MOD,deg[MAX][MAX];
 
 int cmp(Node a,Node b)
 {
     return a.dis<b.dis;
 }
 
 int64 DET(int64 a[][MAX],int n)
 {
     int i,j,k;
     int64 temp=1,t;
     for(i=0;i<n;i++) for(j=0;j<n;j++) a[i][j]%=MOD;
     for(i=1;i<n;i++)
     {
         for(j=i+1;j<n;j++) while(a[j][i])
         {
             t=a[i][i]/a[j][i];
             for(k=i;k<n;k++)
             {
                 a[i][k]-=a[j][k]*t;
                 a[i][k]%=MOD;
             }
             for(k=i;k<n;k++)
             {
                 t=a[i][k];
                 a[i][k]=a[j][k];
                 a[j][k]=t;
             }
             temp=-temp;
         }
         temp=temp*a[i][i]%MOD;
     }
     return (temp+MOD)%MOD;
 }
 
 int64 cal_MST_count()
 {
     sort(e+1,e+m+1,cmp);
     int i,j,k,t,x,y,pre=e[1].dis;
     int64 ans=1;
     a.init(n);
     b.init(n);
     memset(visit,0,sizeof(visit));
     memset(deg,0,sizeof(deg));
     for(i=0;i<=n;i++) g[i].clear();
     for(t=1;t<=m+1;t++)
     {
         if(e[t].dis!=pre||t==m+1)
         {
             for(i=1;i<=n;i++) if(visit[i])
             {
                 k=b.find(i);
                 g[k].push_back(i);
                 visit[i]=0;
             }
             for(i=1;i<=n;i++) if(g[i].size())
             {
                 memset(p,0,sizeof(p));
                 for(j=0;j<g[i].size();j++) for(k=j+1;k<g[i].size();k++)
                 {
                     x=g[i][j];
                     y=g[i][k];
                     p[j][k]=p[k][j]=-deg[x][y];
                     p[j][j]+=deg[x][y];
                     p[k][k]+=deg[x][y];
                 }
                 ans=ans*DET(p,g[i].size())%MOD;
                 for(j=0;j<g[i].size();j++) a.set[g[i][j]]=i;
             }
             memset(deg,0,sizeof(deg));
             for(i=1;i<=n;i++)
             {
                 b.set[i]=a.find(i);
                 g[i].clear();
             }
             if(t==m+1) break;
             pre=e[t].dis;
         }
         x=a.find(e[t].u);
         y=a.find(e[t].v);
         if(x==y) continue;
         visit[x]=visit[y]=1;
         b.Union(x,y);
         deg[x][y]++;
         deg[y][x]++;
     }
     if(!m) return 0;
     for(i=2;i<=n;i++) if(b.find(i)!=b.find(1)) return 0;
     return ans;
 }
 
 int main()
 {
     while(scanf("%d%d%lld",&n,&m,&MOD),n||m||MOD)
     {
         int i;
         for(i=1;i<=m;i++)
         {
             scanf("%d%d%d",&e[i].u,&e[i].v,&e[i].dis);
         }
         int64 x=cal_MST_count();
         printf("%lld\n",x);
     }
     return 0;
 }