杭电 畅通工程 1863 （并查集）

    记录端点数与顶点（村庄）数是否相同，如果相同，所形成的最短路径为最优解输出

 

#include <iostream>#include <cstring>#include <algorithm>using namespace std;#define MAX 5000//  顶点            权      排序    树（并查集）int u[MAX], v[MAX], w[MAX], r[MAX], p[105],  sum, num, n, nc;int cmp(const int a, const int b){    return w[a]<w[b];}int find(int x){    return p[x]==x?x:find(p[x]);}bool Kruskal(){    int i;    for (i=1; i<=num; i++)        p[i] = i;    for (i=1; i<=n; i++)        r[i] = i;    sort(r+1, r+n+1, cmp);    for (i=1; i<=n; i++)    {        int t = r[i];        int x = find(u[t]);        int y = find(v[t]);        if (x!=y)        {            sum += w[t];            p[x] = y;            nc++; //记录端点数        }    }    if (nc==n-1) //端点数=顶点数 返回真        return true;    else        return false;}int main(){    int  i;    while (cin>>num>>n && num)    {        for (i=1; i<=num; i++)            scanf("%d%d%d", &u[i], &v[i], &w[i]);        nc = 0;        sum = 0;        if (Kruskal())            printf("%d\n", sum);        else            printf("?\n");    }    return 0;}