【POJ】1679 The Unique MST

题目链接：http://poj.org/problem?id=1679

题意：给你一组数据，让你判断是否是唯一的最小生成树。

题解：这里用的是kuangbin大佬的次小生成树的模板。直接判断一下次小生成树的最小花费和最小生成树的是否一样即可。

代码：

 #include <iostream>
 #include <cstdio>
 #include <algorithm>
 #include <cstring>
 using namespace std;
 const int maxn = ;
 const int inf = 0x3f3f3f3f;

 bool vis[maxn];
 int lowc[maxn];
 int pre[maxn];
 int Max[maxn][maxn];//Max[i][j]表示在最小生成树中从i到j的路径中的最大边权
 bool used[maxn][maxn];
 int mp[maxn][maxn];
 int n,m;

 int prim(){
     int ans = ;
     memset(vis,false,sizeof(vis));
     memset(Max,,sizeof(Max));
     memset(used,false,sizeof(used));

     vis[] = true;
     pre[] = -;

     for(int i = ; i < n; i++){
         lowc[i] = mp[][i];
         pre[i] = ;
     }

     lowc[] = ;
     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 -;
         ans += minc;
         vis[p] = true;
         used[p][ pre[p] ] = used[ pre[p] ][p] = true;
         for(int j =  ; j < n; j++){
             if(vis[j])
                 Max[j][p] = Max[p][j] = max(Max[j][ pre[p] ],lowc[p]);
             if(!vis[j] && (lowc[j] > mp[p][j] ) ){
                 lowc[j] = mp[p][j];
                 pre[j] = p;
             }
         }
     }
     return ans;
 }
 int ans;
 int smst(){
     int minn = inf;
     for(int i = ; i < n; i++)
         for(int j = i+ ; j < n;j++)
             if(mp[i][j] != inf && !used[i][j]){
                 minn = min(minn, ans + mp[i][j] - Max[i][j]);
             }
     if(minn == inf) return -;//不存在
     return minn;
 }

 int main(){
     int T;
     cin>>T;
     while(T--){
         cin>>n>>m;
         int x,y,w;
         memset(mp,inf,sizeof(mp));
         for(int i = ; i < n ;i++){
             mp[i][i] = ;
         }
         while(m--){
             cin>>x>>y>>w;
             x--,y--;
             mp[x][y] = mp[y][x] = w;
         }
         ans = prim();
         //cout<<smst()<<endl;
         if(ans == -){
             cout<<"Not Unique!"<<endl;
         }
         else if( ans == smst()){
             cout<<"Not Unique!"<<endl;
         }
         else{
             cout<<ans<<endl;
         }
     }

     return ;
 }