【poj1679】The Unique MST

【题目大意】

共T组数据，对于每组数据，给你一个n个点，m条边的图，设图的最小生成树为MST，次小生成树为ans，若MST=ans,输出Not Unique!，否则输出MST

【题解】

很明确，先求MST再求ans。

关于求次小生成树，我打算写一个总结，先留个坑。

 #include<iostream>
 #include<cstdio>
 #include<cstdlib>
 #include<cstring>
 #include<cmath>
 #include<ctime>
 #include<algorithm>
 using namespace std;
 const int INF=~0u>>;//正无穷
 struct node{int x,y,v;}E[];
 struct Node{int y,next,v;}e[];
 int T,n,m,len,mst,ans(INF),Link[],f[],vis[],map[][],flag[][],dp[][];
 inline int read()
 {
     int x=,f=;  char ch=getchar();
     while(!isdigit(ch))  {if(ch=='-')  f=-;  ch=getchar();}
     while(isdigit(ch))  {x=x*+ch-'';  ch=getchar();}
     return x*f;
 }
 void insert(int x,int y,int v)  {e[++len].next=Link[x];Link[x]=len;e[len].y=y;e[len].v=v;}
 int find(int x)  {return f[x]==x?x:f[x]=find(f[x]);}
 bool cmp(node a,node b)  {return a.v<b.v;}
 void Kruskal()
 {
     sort(E+,E+m+,cmp);
     for(int i=;i<=m;i++)
     {
         int x=find(E[i].x),y=find(E[i].y);
         if(x!=y)
         {
             f[x]=y;
             mst+=E[i].v;
             flag[E[i].x][E[i].y]=flag[E[i].y][E[i].x]=;
         }
     }
     for(int i=;i<=n;i++)
         for(int j=i+;j<=n;j++)
             if(flag[i][j])  {insert(i,j,map[i][j]);  insert(j,i,map[j][i]);}
 }
 void dfs(int st,int x,int v)
 {
     vis[x]=;
     dp[st][x]=v;
     for(int i=Link[x];i;i=e[i].next)
         if(!vis[e[i].y])  dfs(st,e[i].y,max(dp[st][x],e[i].v));
 }
 void pre()
 {
     memset(Link,,sizeof(Link));
     memset(flag,,sizeof(flag));
     memset(dp,,sizeof(dp));
     len=mst=;  ans=INF;
     for(int i=;i<=n;i++)  f[i]=i;
     for(int i=;i<=n;i++)
         for(int j=;j<=n;j++)
             map[i][j]=(i==j?:INF);
 }
 int main()
 {
     freopen("cin.in","r",stdin);
     freopen("cout.out","w",stdout);
     T=read();
     while(T--)
     {
         n=read();  m=read();
         pre();
         for(int i=;i<=m;i++)
         {
             int x=read(),y=read(),v=read();
             map[x][y]=map[y][x]=v;
             E[i].x=x;  E[i].y=y;  E[i].v=v;
         }
         Kruskal();
         for(int i=;i<=n;i++)  {memset(vis,,sizeof(vis)); dfs(i,i,);}
         for(int i=;i<=n;i++)
             for(int j=i+;j<=n;j++)
                 if(!flag[i][j]&&map[i][j]!=INF)  ans=min(ans,mst+map[i][j]-dp[i][j]);
         if(ans==mst)  printf("Not Unique!\n");
         else printf("%d\n",mst);
     }
     return ;
 }