06-图6. 公路村村通(30)
时间限制
400 ms
内存限制
65536 kB
代码长度限制
8000 B
判题程序
Standard
现有村落间道路的统计数据表中,列出了有可能建设成标准公路的若干条道路的成本,求使每个村落都有公路连通所需要的最低成本。
输入格式说明:
输入数据包括城镇数目正整数N(<=1000)和候选道路数目M(<=3N);随后的M行对应M条道路,每行给出3个正整数,分别是该条道路直接连通的两个城镇的编号以及该道路改建的预算成本。为简单起见,城镇从1到N编号。
输出格式说明:
输出村村通需要的最低成本。如果输入数据不足以保证畅通,则输出-1,表示需要建设更多公路。
样例输入与输出:
序号 输入 输出
1
6 15
1 2 5
1 3 3
1 4 7
1 5 4
1 6 2
2 3 4
2 4 6
2 5 2
2 6 6
3 4 6
3 5 1
3 6 1
4 5 10
4 6 8
5 6 3
12
2
3 1
2 3 2
-1
3
5 4
1 2 1
2 3 2
3 1 3
4 5 4
**这是一个最小生成树Kruskal算法的题,并不难,先把边数用一个结构体存储下来,然后用快排对其按权值排序,然后依次按照并查集的做法放在一个集合里,最后判断一下,如果不是一个集合,那就是没有连通,输出-1,否则输出SUM
复习并查集知识,这里并没有对并查集的时间加以限制,因此在并的时候即使不是最优的也会AC,并查集的操作有:
1,找到两个点的根节点。
2,如果两个根节点不相等,那就把一个节点 靠在另一个上。
3,注意把小的集合挂在大的上。**
#include<cstdio>
#include<iostream>
#include<algorithm>
//#define LOCAL
using namespace std;
typedef struct {
int x,y;
int weight;
}Edge;
int m,n;
int cmp(Edge a,Edge b);
void Union(int* s,int* p,Edge a,int* sum);
int Find(int* s,int* p,int k);
int main(){
#ifdef LOCAL
freopen("data.in","r",stdin);
freopen("data.out","w",stdout);
#endif
cin >> m >> n ;
Edge edge[n];
int i;
for(i = 0;i<n;i++)
scanf("%d%d%d",&edge[i].x,&edge[i].y,&edge[i].weight);
// cin >> edge[i].x >> edge[i].y >> edge[i].weight;
int k = n;
// 用快排从小到大排序
sort(edge,edge+k,cmp);
// 并查集初始化
int Gather[m+1],parent[m+1],sum = 0;
for(i = 1;i<m+1;i++){
parent[i] = -1;
Gather[i] = i;
}
// 构建集合
for(i = 0;i<n;i++)
Union(Gather,parent,edge[i],&sum);
// 判断是否所有点都连上
int p = 0;
for(i = 1;i<m+1;i++){
if(parent[i]<0)
p++;
if(p>1)
break;
}
if(p>1)
cout << -1;
else
cout << sum;
return 0;
}
int cmp(Edge a,Edge b){
return a.weight<b.weight;
}
void Union(int* s,int* p,Edge a,int* sum){
int root1,root2;
root1 = Find(s,p,a.x);
root2 = Find(s,p,a.y);
if(root1 != root2){
// p[root1]比p[root2]大
if(p[root1] < p[root2]){
p[root1] += p[root2];
p[root2] = root1;
*sum += a.weight;
}
// p[root1]比p[root2]小
else{
p[root2] += p[root1];
p[root1] = root2;
*sum += a.weight;
}
}
}
int Find(int* s,int* p,int k){
int i;
for(i = 1;i<m+1&&s[i]!=k;i++);
if(i>=m+1)
return -1;
for(;p[i]>0;i = p[i]);
return i;
}