poj 1679 The Unique MST 次小生成树算法

#include <stdio.h>
#include <string.h>
int map[110][110],max[110][110],p[110],low[110],n,m,vis[110][110];
/*max[i][j]表示最小生成树中从i到j的路径中最长的边，map是边，p[i]是prim每次选进去的边一个节点（在之前选进去的集合中的节点）
vis[i][j]若为0，表示i到j无边，等于一表示有边但未选入最小生成树，等于2表示选入了最小生成树*/
int MAX(int a,int b)
{
if(a>b)
return a;
else return b;
}
int prim(int u)
{
int i,min,v,j,cost=0;
for(j=1;j<=n;j++)
{
low[j]=map[j][u];
p[j]=u;
}
low[u]=-1;
for(j=1;j<=n-1;j++)
{
min=50000;
for(i=1;i<=n;i++)
{
if(low[i]!=-1&&min>low[i])
{
min=low[i];
v=i;
}
}
low[v]=-1;
cost+=min;
vis[v][p[v]]=vis[p[v]][v]=2;/*在生成树中，所以改为2*/
for(i=1;i<=n;i++)
if(i!=v)
max[i][v]=max[v][i]=MAX(min,max[i][p[v]]);/*每次选入一个边都更新一遍max*/
for(i=1;i<=n;i++)
{
if(low[i]!=-1&&map[v][i]<low[i])
{
low[i]=map[v][i];
p[i]=v;
}
}
}
return cost;
}

int main()
{
int t,i,j,k,x,y,min,sum,w;
scanf("%d",&t);
for(i=0;i<t;i++)
{
min=50000;
scanf("%d%d",&n,&m);
memset(max,0,sizeof(max));/*刚开始时初始为0*/
memset(vis,0,sizeof(vis));
for(j=1;j<=n;j++)
for(k=1;k<=n;k++)
map[j][k]=50000;
for(j=0;j<m;j++)
{
scanf("%d%d%d",&x,&y,&w);
map[x][y]=map[y][x]=w;
vis[x][y]=vis[y][x]=1;
}
sum=prim(1);
for(j=1;j<=n;j++)
for(k=1;k<=n;k++)
if(vis[j][k]==1&&min>sum+map[j][k]-max[j][k])
/*将不在生成树中的边依次加入生成树，形成一个环，去掉环上原生成树中的最大边形成一棵较大的生成树*/
min=sum+map[j][k]-max[j][k];
if(min!=sum)
printf("%d\n",sum);
else
printf("Not Unique!\n");
}
return 0;
}