hdu 1233 还是畅通工程

http://acm.hdu.edu.cn/showproblem.php?pid=1233


这个题我用的贪心和并查集来解决的。 (kruskal算法)
首先用贪心对长度排序,然后按照长度从小到大选择连接,没有被联通的则直接连,要注意一个问题就是如果4个村庄是1 4联通了, 2 3联通了,那么肯定还要让这两个集合联通(用并查集)。
因此我用了一个p来表示不同的集合。

AC代码:
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>

using namespace std;

struct Node
{
    int a,b;
    int l;
}a[5010];

bool cmp(Node a, Node b)
{
    return a.l < b.l;
}

int main()
{
    int n,i,j,sum,p,t;
    int b[110];       //记录这个村庄是否已经联通
    while(scanf("%d",&n)&&n)
    {
        sum = 0;                  //初始化总费用
        memset(b,0,sizeof(b));    //初始化所有村庄不连通
        for(i = 0; i < n*(n-1)/2; i++)
        {
            scanf("%d%d%d",&a[i].a,&a[i].b,&a[i].l);
        }
        sort(a,a+n*(n-1)/2,cmp); //权值从小到大排序
        p = 1;
        for(i = 0; i < n*(n-1)/2; i++)
        {
            if(b[a[i].a] == 0 || b[a[i].b] == 0)
            {
                sum += a[i].l;
                if(b[a[i].a] == 0 && b[a[i].b] == 0) //如果两个都没有被联通过,则为新联通图集合
                {
                    b[a[i].a] = b[a[i].b] = p;
                    p++;
                }
                else
                {
                    if(b[a[i].a] != 0)
                    {
                        b[a[i].b] = b[a[i].a];
                    }
                    else
                    {
                        b[a[i].a] = b[a[i].b];
                    }
                }
            if(b[a[i].a] != b[a[i].b])   //连个村庄在两个图的时候
            {
                t = b[a[i].b];
                sum += a[i].l;
                for(j = 1; j <= n; j++) //两集合合并
                {
                    if(b[j] == t)
                    {
                        b[j] = b[a[i].a];
                    }
                }
            }

        }
        printf("%d
",sum);
    }

    return 0;
}


原文地址:https://www.cnblogs.com/dyllove98/p/3236906.html