当前位置: 代码迷 >> 综合 >> HDU 1233 还是畅通工程 (并查集)
  详细解决方案

HDU 1233 还是畅通工程 (并查集)

热度:64   发布时间:2023-12-05 06:32:38.0
#include <bits/stdc++.h>
using namespace std;
const int maxn=10005; 
struct node
{int f;int l;int len;
}a[maxn];//定义一个结构体储存公路的信息 
int n,m,b[maxn],ans;
bool cmp(node a,node b)
{return a.len<b.len;//按公路的长度从小到大排序 
}
int find(int x)
{int r=x;while(b[r]!=r)//寻找父节点 r=b[r];//如果当前部位父节点则继续寻找 return r;//返回父节点 
}
void mergy(int x,int y)
{int fx=find(x);int fy=find(y);if(fx!=fy)b[fy]=x;//将父节点合并 
}
int kruskal()
{sort(a,a+m,cmp);//排序,优先查找长度小的 for(int i=0;i<=n;i++)b[i]=i;//给b数组赋值 ans=0;//长度总和 for(int i=0;i<m;i++)if(find(a[i].f)!=find(a[i].l))//如果两个村子没有公路 {mergy(a[i].f,a[i].l);//查找父节点 ans+=a[i].len;//长度相加	}return ans;
}
int main()
{while(scanf("%d",&n)!=EOF&&n!=0){m=n*(n-1)/2;for(int i=0;i<m;i++)scanf("%d %d %d",&a[i].f,&a[i].l,&a[i].len);		printf("%d\n",kruskal());}return 0;
}