hdu 1879 继续畅通工程 解题报告

题目链接：http://code.hdu.edu.cn/showproblem.php?pid=1879

      这条题目我的做法与解决Constructing Roads的解法是相同的。

      0 表示没有连通；1代表已连通。在已连通村庄的道路的基础上，找到扩展出来的最小生成树。

     

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

const int maxe = 10000 + 10;
int rep[maxe], vis[maxe];

struct sets
{
    int v1, v2;
    int value;
} exa[maxe];

int cmp(sets a, sets b)
{
    return a.value < b.value;
}

int find(int x)
{
    return x == rep[x] ? x : find(rep[x]);
}

int main()
{
    int i, n, a, b, c, x, y, cnt, flag, sign;
    while (scanf("%d", &n) != EOF && n)
    {
        for (i = 1; i <= n; i++)
            rep[i] = i;
        cnt = flag = 0;
        n *= (n-1)/2; 
        memset(vis, 0, sizeof(vis));
        for (i = 0; i < n; i++)
        {
           scanf("%d%d%d%d", &a, &b, &c, &sign);
           if (a > b)
               swap(a, b);
           if (sign == 0)
           {
               exa[cnt].v1 = a;
               exa[cnt].v2 = b;
               exa[cnt].value = c;
               cnt++;
            }
           else
           {
               x = find(a);    
               y = find(b);
               if (x > y)
                   swap(x, y);
               if (vis[a] && vis[b])
                   rep[x] = y;
               else
               {
                   if (x == y)
                       flag = 1;  
                   if (!vis[a] && !vis[b])
                   {    
                       rep[x] = y;
                       vis[a] = vis[b] = 1;
                    }
                   else if (!vis[a])
                   {
                       rep[x] = y;
                       vis[a] = 1;
                   }
                   else if (!vis[b])
                   {
                       rep[x] = y;
                       vis[b] = 1;
                   }
               } 
           }
        }
        sort(exa, exa+cnt, cmp);
        if (!flag)
        {
            int minval = 0;
            for (i = 0; i < cnt; i++)
            {
                x = find(exa[i].v1);
                y = find(exa[i].v2);
                if (x != y)
                {
                    minval += exa[i].value;
                    if (x < y)
                        rep[x] = y;
                    else
                        rep[y] = x;
                }
            }    
            printf("%d\n", minval);
        }
        else
            printf("0\n");
    }
    return 0;
}