您的位置:首页 > 产品设计 > UI/UE

POJ 1679-The Unique MST(次小生成树)

2016-11-08 16:57 435 查看
题意:

找出在给出的图中是否存在两个最小生成树,如果有输出不独立,否则如果只有一个就输出最小生成树的权值。

模板题:

最小生成树的模板

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <cstring>
using namespace std;
#define maxn 111
const int inf =0x3f3f3f3f;
int maxx[maxn][maxn];
int mp[maxn][maxn];
int vis[maxn];
int dis[maxn];
bool connect[maxn][maxn];
int father[maxn];
int n,m;
int prim()
{
memset(vis,false,sizeof(vis));
for(int i=1;i<=n;i++)
{
father[i]=i;
dis[i]=mp[1][i];
father[i]=1;
}
vis[1]=1;
dis[1]=0;
int res=0;
for(int i=1;i<=n;i++)
{
int index=-1,minn=inf;
for(int j=1;j<=n;j++)
{
if(vis[j]==0&&dis[j]<minn)
{
index=j;
minn=dis[j];
}
}
if(index==-1)
{
return res;
}
int pre=father[index];
vis[index]=1;
res+=minn;
connect[pre][index]=false;
connect[index][pre]=false;
maxx[pre][index]=minn;
for(int j=1;j<=n;j++)
maxx[j][index]=max( maxx[pre][index],maxx[j][pre] );
for(int j=1;j<=n;j++)
{
if(vis[j]==0&&dis[j]>mp[index][j])
{
dis[j]=mp[index][j];
father[j]=index;
}
}
}
return res;
}
int main()
{
int T;
cin>>T;
while(T--)
{
memset(mp,inf,sizeof(mp));
memset(connect,false,sizeof(connect));
cin>>n>>m;
for(int i=1;i<=m;i++)
{
int x,y,d;
scanf("%d%d%d",&x,&y,&d);
mp[x][y]=d;
mp[y][x]=d;
connect[x][y]=true;
connect[y][x]=true;
}
int ans=prim();
int over=0;
for(int i=1;!over&&i<=n;i++)
{
for(int j=1;j<=n;j++)
{
if(connect[i][j]==false||mp[i][j]==inf)
continue;
if(mp[i][j]==maxx[i][j])
{

4000
over=1;
break;
}
}
}

if(over)
printf("Not Unique!\n");
else
printf("%d\n",ans);
}
return 0;
}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: