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;

 }