首页 > 代码库 > poj 1679 The Unique MST 最小生成树
poj 1679 The Unique MST 最小生成树
点击打开链接 http://poj.org/problem?id=1679
题意:给一个无向图,问最小生成树是否唯一,如果唯一就输出最小生成树的所有边的权值的和,如果不唯一,那么就输出Not Unique!
思路:在用prime算法求最小生成树的过程中,在找到权值最小的一个节点之后,先判断一下,这个节点的权值是否可以由好几条路径求得,并且权值都等于当前权值,如果是的话,那么最小生成树就不唯一。那么如何判断呢?可以用一个flag数组标记,0代表这个点得到当前权值的路径唯一,1代表不唯一,初始化全为0。我们在用一个已找到的权值最小的节点i来更新其他点的权值的时候,如果发现某个点的现在的权值和i点到这个点的距离相等,那么就说明这个点最少可以由两条路径来得到当前的这个最小值,就把他标记为1,如果这个点的权值比i点到它的距离要小,那么就更行这个点的权值,并把flag标记为0,因为当前的这个最小值只能由i点得到,是唯一的,剩下的就是一个简单的最小生成树问题了。
#include<cstdio> #include<cstring> #include<iostream> using namespace std; const int maxn=110; struct Side{ int to,v,next; }side[2*maxn*maxn]; int head[maxn],top; int dis[maxn]; int n,m; bool vis[maxn],flag[maxn]; void init(){ memset(head,-1,sizeof(head)); top=0; memset(vis,0,sizeof(vis)); memset(dis,0x3f3f,sizeof(dis)); memset(flag,0,sizeof(flag)); } void addside(int a,int b,int c){ side[top]=(Side){b,c,head[a]}; head[a]=top++; side[top]=(Side){a,c,head[b]}; head[b]=top++; } bool prime(){ vis[1]=1; dis[1]=0; for(int i=head[1];i!=-1;i=side[i].next){ dis[side[i].to]=side[i].v; } for(int i=0;i<n-1;i++){ int j,minc=1e9; for(int k=1;k<=n;k++){ if(!vis[k]&&dis[k]<minc){ minc=dis[k]; j=k; } } if(!flag[j]) vis[j]=1; else return 0; for(int k=head[j];k!=-1;k=side[k].next){ int to=side[k].to; int v=side[k].v; if(vis[to]) continue; if(dis[to]>v){ flag[to]=0; dis[to]=v; } else if(dis[to]==v) flag[to]=1; } } return 1; } int main() { int cas; cin>>cas; while(cas--){ init(); cin>>n>>m; int a,b,c; for(int i=0;i<m;i++){ scanf("%d %d %d",&a,&b,&c); addside(a,b,c); } if(prime()){ int ans=0; for(int i=1;i<=n;i++) ans+=dis[i]; cout<<ans<<endl; } else cout<<"Not Unique!"<<endl; } return 0; }
poj 1679 The Unique MST 最小生成树
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。