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;
}