poj1679TheUniqueMST次小生成树

SoL：裸的次小生成树。。。

推荐：KuangBin巨巨的博客 模版 &＃43; 详解      http://www.cnblogs.com/kuangbin/p/3147329.html

#include
#include
#include
#include
using namespace std;
const int maxn = 100 + 10;
const int INF = 0x3f3f3f3f;
int ans;
bool vis[maxn];
int lowc[maxn];
int pre[maxn];
int Max[maxn][maxn];
bool used[maxn][maxn];
int Prim(int cost[][maxn],int n)
{
int ans=0;
memset(vis,false,sizeof(vis));
memset(Max,0,sizeof(Max));
memset(used,false,sizeof(used));
vis[0]=true;
pre[0]=-1;
for(int i=1;i {
lowc[i]=cost[0][i];
pre[i]=0;
}
lowc[0]=0;
for(int i=1;i {
int minc=INF;
int p=-1;
for(int j=0;j if(!vis[j]&&minc>lowc[j])
{
minc=lowc[j];
p=j;
}
if(minc==INF) return -1;
ans+=minc;
vis[p]=true;
used[p][pre[p]]=used[pre[p]][p]=true;
for(int j=0;j {
if(vis[j])
Max[j][p]=Max[p][j]=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;
}
int smst(int cost[][maxn],int n)
{
int Min=INF;
for(int i=0;i for(int j=i+1;j if(cost[i][j]!=INF&&!used[i][j])
Min=min(Min,ans+cost[i][j]-Max[i][j]);
if(Min==INF) return -1;
return Min;
}
int cost[maxn][maxn];
int T,n,m;
int main()
{
scanf("%d",&T);
while(T--)
{
int u,v,w;
scanf("%d%d",&n,&m);
for(int i=0;i for(int j=0;j {
if(i==j) cost[i][j]=0;
else cost[i][j]=INF;
}
while(m--)//0开始
{
scanf("%d%d%d",&u,&v,&w);
u--,v--;
cost[u][v]=cost[v][u]=w;
}
ans=Prim(cost,n);
if(ans==-1)
{
printf("Not Unique!\n");
continue;
}
if(ans==smst(cost,n))
printf("Not Unique!\n");
else
printf("%d\n",ans);
}
return 0;
}