Time Limit: 4000/2000 MS (Java/Others) Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 67160 Accepted Submission(s): 30440
#include<iostream> #include<algorithm> #include<string.h> using namespace std; int a[105][105],dis[105],vis[105];//vis标记处理过的点,dis[i]表示从起点到i点的权值 int n,ans; void prim() { ans=0; memset(vis,0,sizeof(vis)); for(int i=2;i<=n;i++)//初始化 dis[i]=a[1][i]; dis[1]=0; vis[1]=1; for(int i=1;i<n;i++)//最后一个点不需要处理,直接加入即可,所以不要(也不能)取等 { int k=0,mn=9999999; for(int j=1;j<=n;j++)//找出还没有被标记的点中离起点权值最小的点 { if(!vis[j]&&dis[j]<mn) { mn=dis[j]; k=j; } } vis[k]=1; ans=ans+mn; for(int j=1;j<=n;j++)//更新最小值,k和起点都在处理过的集合里面,更新到起点的最小值 { if(!vis[j]&&dis[j]>a[k][j]) dis[j]=a[k][j]; } } } int main() { while(~scanf("%d",&n)&&n) { int x,y,v; for(int i=1;i<=n*(n-1)/2;i++) { cin>>x>>y>>v; a[x][y]=a[y][x]=v; } prim(); printf("%d\n",ans); } }
#include<iostream> #include<algorithm> using namespace std; int p[105],r[105]; int n,ans; struct node { int x; int y; int v; }a[100005]; bool cmp(node b,node c) { return b.v<c.v; } int find(int x)//查找元素x的老板是谁 { if (x == p[x]) return x; else return p[x] = find(p[x]); } void join(int x, int y)//合并两个集合 { int xRoot = find(x); int yRoot = find(y); if (xRoot == yRoot) //老板相同,不合并 return; //cnt=cnt-1; if (r[xRoot] < r[yRoot]) //r[i]是元素i所在树的高度,矮树的根节点认高树的根节点做老板 p[xRoot] = yRoot; else if (r[xRoot] > r[yRoot]) p[yRoot] = xRoot; else { p[yRoot] = xRoot;//树高相同,做老板的树高度要加一 r[xRoot]++; } } void kruskal() { for(int i=1;i<=n;i++)//初始化根节点 p[i]=i; sort(a+1,a+n*(n-1)/2+1,cmp); for(int i=1;i<=n*(n-1)/2;i++) { if(find(a[i].x)!=find(a[i].y)) { join(a[i].x,a[i].y); ans=ans+a[i].v; } } } int main() { while(~scanf("%d",&n)) { ans=0; if(n==0) break; else { for(int i=1;i<=n*(n-1)/2;i++) scanf("%d %d %d",&a[i].x,&a[i].y,&a[i].v); //cin>>a[i].x>>a[i].y>>a[i].v; kruskal(); } printf("%d\n",ans); //cout<<ans<<endl; } return 0; }
原文:https://www.cnblogs.com/-citywall123/p/10999949.html