[ An Ac a Day ^_^ ][kuangbin带你飞]专题八 生成树 POJ 1679 The Unique MST

求最小生成树是否唯一

求一遍最小生成树再求一遍次小生成树 看看值是否相等就可以

 #include<cstdio>
 #include<iostream>
 #include<algorithm>
 #include<cmath>
 #include<cstring>
 #include<string>
 #define cl(a,b) memset(a,b,sizeof(a))
 #define debug(x) cerr<<#x<<"=="<<(x)<<endl
 using namespace std;
 typedef long long ll;

 const int maxn=1e2+;
 const int inf=0x3f3f3f3f;

 int n,m;
 int cost[maxn][maxn];
 int pre[maxn],lowc[maxn];
 int Max[maxn][maxn];
 bool vis[maxn],used[maxn][maxn];

 int mst()
 {
     int ans=;
     cl(vis,false),cl(Max,),cl(used,false);
     vis[]=true,pre[]=-,lowc[]=;
     for(int i=;i<=n;i++)
     {
         lowc[i]=cost[][i];
         pre[i]=;
     }
     for(int i=;i<=n;i++)
     {
         int minc=inf;
         int p=-;
         for(int j=;j<=n;j++)
         {
             if(!vis[j]&&minc>lowc[j])
             {
                 minc=lowc[j];
                 p=j;
             }
         }
 //        if(minc==inf) return -1;
         ans+=minc;
         vis[p]=used[p][pre[p]]=used[pre[p]][p]=true;
         for(int j=;j<=n;j++)
         {
             if(vis[j])
             {
                 Max[j][p]=max(Max[j][pre[p]],lowc[p]);
             }
             if(!vis[j]&&lowc[j]>cost[p][j])
             {
                 lowc[j]=cost[p][j];
                 pre[j]=p;
             }
         }
     }
     return ans;
 }

 void solve()
 {
     int ans=mst();
     int Min=inf;
     for(int i=;i<=n;i++)
     {
         for(int j=i+;j<=n;j++)
         {
             if(cost[i][j]!=inf&&!used[i][j])
             {
                 Min=min(Min,ans+cost[i][j]-Max[i][j]);
             }
         }
     }
     if(ans==Min) printf("Not Unique!\n");
     else printf("%d\n",ans);
 }

 int main()
 {
     int T;
     scanf("%d",&T);
     while(T--)
     {
         cl(cost,inf);
         scanf("%d%d",&n,&m);
         for(int i=;i<m;i++)
         {
             int u,v,w;
             scanf("%d%d%d",&u,&v,&w);
             cost[u][v]=cost[v][u]=w;
         }
         solve();
     }
     return ;
 }