POJ 1287 Networking（最小生成树kruskal）

Networking

题目链接：

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

解题思路：

题目大意：

给你p个节点，和r条边，这r条边代表从u节点到v节点电缆的长度，现在要你将所有节点都连起来，并且使长度最小。

最小生成树裸题。

AC代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;

const int maxn = 105;
int p,r;
int pa[55];

struct Edge{
int u,v,w;
}edge[maxn];

bool cmp(Edge a,Edge b){
return a.w < b.w;
}

int findset(int x){
if(pa[x] != x)
pa[x] = findset(pa[x]);
return pa[x];
}

int kruskal(){
int cnt = p,sum = 0;
for(int i = 0; i < r; i++){
int u = findset(edge[i].u);
int v = findset(edge[i].v);
if(u != v){
sum += edge[i].w;
pa[v] = u;
if(--cnt == 0)
break;
}
}
return sum;
}

int main(){
while(scanf("%d",&p),p){
for(int i = 0; i <= p; i++)
pa[i] = i;
scanf("%d",&r);
int u,v,w;
for(int i = 0; i < r; i++)
scanf("%d%d%d",&edge[i].u,&edge[i].v,&edge[i].w);
sort(edge,edge+r,cmp);
printf("%d\n",kruskal());
}
return 0;
}