POJ 1679 The Unique MST 确定MST是否唯一

  1 #include<cstdio>
  2 #include<cstring>
  3 #include<algorithm>
  4 
  5 using namespace std;
  6 
  7 const int MAXN=103;
  8 const int INF=0x3f3f3f3f;
  9 
 10 int cost[MAXN][MAXN];
 11 int pre[MAXN];
 12 int lowc[MAXN];
 13 bool vis[MAXN];
 14 
 15 void init(int N)
 16 {
 17     for(int i=1;i<=N;i++)
 18     {
 19         for(int j=1;j<=N;j++)
 20         {
 21             if(i==j)
 22                 cost[i][j]=0;
 23             else
 24                 cost[i][j]=INF;
 25         }
 26     }
 27 }
 28 
 29 int prim(int N,int cnt)
 30 {
 31     memset(vis,false,sizeof(vis));
 32     if(cnt)
 33         memset(pre,-1,sizeof(pre));
 34     vis[1]=true;
 35     int ans=0;
 36     for(int i=1;i<=N;i++)
 37     {
 38         lowc[i]=cost[1][i];
 39         if(cnt)
 40             pre[i]=1;
 41     }
 42 
 43     for(int j=1;j<N;j++)
 44     {
 45         int p=-1;
 46         int minc=INF;
 47         for(int i=1;i<=N;i++)
 48         {
 49             if(!vis[i]&&lowc[i]<minc)
 50             {
 51                 minc=lowc[i];
 52                 p=i;
 53             }
 54         }
 55         if(p==-1)
 56             return -1;
 57         ans+=minc;
 58         vis[p]=true;
 59         for(int i=1;i<=N;i++)
 60         {
 61             if(!vis[i]&&cost[p][i]<lowc[i])
 62             {
 63                 lowc[i]=cost[p][i];
 64                 if(cnt)
 65                     pre[i]=p;
 66             }
 67         }
 68     }
 69     return ans;
 70 }
 71 
 72 int main()
 73 {
 74     int test;
 75     scanf("%d",&test);
 76     while(test--)
 77     {
 78         int N,M;
 79         scanf("%d%d",&N,&M);
 80         init(N);
 81 
 82         for(int i=0;i<M;i++)
 83         {
 84             int u,v,w;
 85             scanf("%d%d%d",&u,&v,&w);
 86             cost[u][v]=cost[v][u]=w;
 87         }
 88 
 89         int ans=prim(N,1);
 90 
 91         for(int i=2;i<=N;i++)
 92         {
 93             int v=pre[i];
 94             int tmp=cost[i][v];
 95             cost[i][v]=cost[v][i]=INF;
 96             int sum=prim(N,0);
 97             if(ans==sum)
 98             {
 99                 ans=-1;
100                 break;
101             }
102             cost[i][v]=cost[v][i]=tmp;
103         }
104         if(ans==-1)
105             printf("Not Unique!\n");
106         else
107             printf("%d\n",ans);
108     }
109     return 0;
110 }