[ An Ac a Day ^_^ ][kuangbin带你飞]专题八 生成树 POJ 1679 The Unique MST

求最小生成树是否唯一

求一遍最小生成树再求一遍次小生成树 看看值是否相等就可以

 #include<cstdio>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<cstring>
#include<string>
#define cl(a,b) memset(a,b,sizeof(a))
#define debug(x) cerr<<#x<<"=="<<(x)<<endl
using namespace std;
typedef long long ll; const int maxn=1e2+;
const int inf=0x3f3f3f3f; int n,m;
int cost[maxn][maxn];
int pre[maxn],lowc[maxn];
int Max[maxn][maxn];
bool vis[maxn],used[maxn][maxn]; int mst()
{
int ans=;
cl(vis,false),cl(Max,),cl(used,false);
vis[]=true,pre[]=-,lowc[]=;
for(int i=;i<=n;i++)
{
lowc[i]=cost[][i];
pre[i]=;
}
for(int i=;i<=n;i++)
{
int minc=inf;
int p=-;
for(int j=;j<=n;j++)
{
if(!vis[j]&&minc>lowc[j])
{
minc=lowc[j];
p=j;
}
}
// if(minc==inf) return -1;
ans+=minc;
vis[p]=used[p][pre[p]]=used[pre[p]][p]=true;
for(int j=;j<=n;j++)
{
if(vis[j])
{
Max[j][p]=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;
} void solve()
{
int ans=mst();
int Min=inf;
for(int i=;i<=n;i++)
{
for(int j=i+;j<=n;j++)
{
if(cost[i][j]!=inf&&!used[i][j])
{
Min=min(Min,ans+cost[i][j]-Max[i][j]);
}
}
}
if(ans==Min) printf("Not Unique!\n");
else printf("%d\n",ans);
} int main()
{
int T;
scanf("%d",&T);
while(T--)
{
cl(cost,inf);
scanf("%d%d",&n,&m);
for(int i=;i<m;i++)
{
int u,v,w;
scanf("%d%d%d",&u,&v,&w);
cost[u][v]=cost[v][u]=w;
}
solve();
}
return ;
}
上一篇:使用Nodejs实现实时推送MySQL数据库最新信息到客户端


下一篇:基于node.js 的 websocket的移动端H5直播开发