POJ 1679

求一次最小成树，求一次小生成树，若相等，则不唯一。否则，唯一。

 #include <iostream>
 #include <cstdio>
 #include <cstring>
 #include <algorithm>

 using namespace std;
 const int MAXN=;
 const int inf=;
 int map[MAXN][MAXN]; int f[MAXN][MAXN];
 bool exist[MAXN][MAXN],used[MAXN][MAXN];
 int least[MAXN],pre[MAXN]; bool vis[MAXN];
 int n,m,ans1,ans2;

 void prim(){
     memset(f,,sizeof(f));
     least[]=;
     for(int i=;i<=n;i++){
         least[i]=map[][i];
         if(least[i]!=inf)
         pre[i]=;
     }
     vis[]=true;
     for(int i=;i<=n;i++){
         int min=inf,p=-;
         for(int k=;k<=n;k++){
             if(least[k]<min&&!vis[k]){
                 p=k; min=least[k];
             }
         }

         if(p==-) return ;
         ans1+=min;
         used[pre[p]][p]=used[p][pre[p]]=;
         for(int j=;j<=n;j++){
         if(vis[j])
         f[j][p]=max(f[j][pre[p]],map[pre[p]][p]);
         f[p][j]=f[j][p];
         }
         vis[p]=true;
         for(int k=;k<=n;k++){
             if(!vis[k]&&map[p][k]<least[k]){
                 least[k]=map[p][k];
                 pre[k]=p;
             }
         }
     }
 }

 void secondT(){
     ans2=inf;
     for(int i=;i<=n;i++){
         for(int j=;j<=n;j++){
             if(exist[i][j]&&!used[i][j]&&ans1 + map[i][j] - f[i][j] < ans2)
             ans2=ans1 + map[i][j] - f[i][j];
         }
     }
 }

 int main(){
     int T;
     scanf("%d",&T);
     while(T--){
         scanf("%d%d",&n,&m);
         int u,v,d;
         ans1=ans2=;
         memset(exist,-,sizeof(exist));
         for(int i=;i<=n;i++){
             for(int j=i;j<=n;j++)
             map[i][j]=map[j][i]=inf;
         }
         for(int i=;i<=m;i++){
             scanf("%d%d%d",&u,&v,&d);
             map[u][v]=map[v][u]=d;
             exist[u][v]=exist[v][u]=;
         }
         memset(pre,-,sizeof(pre));
         memset(used,,sizeof(used));
         memset(vis,false,sizeof(vis));
         prim();
         secondT();
         if(ans1==ans2)
         printf("Not Unique!\n");
         else printf("%d\n",ans1);
     }
     return ;
 }