poj 1679 判断最小生成树是否唯一

http://poj.org/problem?id=1679

题意：判断最小生成树是否唯一，唯一输出该值，否则为“Not Unique!”

分析：这里用Kruskal算法做最小生成树，遍历每条在最小生成树上的边，做N-1次最小生成树 (代码很搓，时间效率也不太好)

//kruskal算法 #include <stdio.h> #include <string.h> #include <stdlib.h> struct e { int u,v,w; }edge[10010]; int mst_edge[10010]; int father[110]; int n,m; int mst=0,sec_mst; int cmp(const void *a,const void *b) { return ((struct e *)a)->w - ((struct e *)b)->w; } void init(void) { for(int i=1;i<105;i++) father[i]=i; } int find(int x) { if(x != father[x]) father[x]=find(father[x]); return father[x]; } int kruskal(int flag=-1) { init(); int ans=0; int cnt=1; if(flag == -1) { for(int i=1;i<=m;i++) { if(cnt == n) break; int x=find(edge[i].u); int y=find(edge[i].v); if(x!=y) { mst_edge[cnt]=i; ans+=edge[i].w; father[x]=y; cnt++; } } return ans; } else { for(int i=1;i<=m;i++) { if(cnt == n) break; if(i == mst_edge[flag]) continue; int x=find(edge[i].u); int y=find(edge[i].v); if(x!=y) { ans+=edge[i].w; father[x]=y; cnt++; } } int temp=0; for(int j=1;j<=n;j++) if(father[j]==j) temp++; if(temp>1) return -1; return ans; } } int main() { int cas; scanf("%d",&cas); while(cas--) { memset(mst_edge,0,sizeof(mst_edge)); memset(edge,0,sizeof(edge)); sec_mst=0xfffffff; scanf("%d%d",&n,&m); for(int i=1;i<=m;i++) scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w); if(n == 1) { printf("0/n"); continue; } qsort(edge+1,m,sizeof(edge[1]),cmp); mst=kruskal(); for(int j=1;j<=n-1;j++) { int res; res=kruskal(j); if(res<=sec_mst && res>0) sec_mst=res; } if(mst == sec_mst) printf("Not Unique!/n"); else printf("%d/n",mst); } return 0; }

转载请注明，谢谢。