HDU 1102 Constructing Roads

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

题意：

有N个村庄，编号从1到N，你应该建立一些道路，使每两个村庄可以连接到彼此。我们说两个村庄A和B连接，当且仅当在A和B之间有道路，或者存在一个村庄C，使得在A和C之间有道路，并且C和B连接。
我们知道，一些村庄之间已经有一些道路，你的工作是建造一些道路，使所有的村庄都连接，所有建设的道路的长度是最小的。

 

思路：

普里姆算法或克鲁斯卡尔算法。

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

const int maxn = 100 + 5;

int n, cnt, ans;

//边
struct node
{
    int u;    //边的起点
    int v;    //边的终点
    int w;    //边的权重
}edge[maxn*maxn];

int p[maxn];

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

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

void Kruskal()
{
    //首先按权重从小到大排序
    sort(edge, edge + cnt, cmp); 
    //依次选择权重小的
    for (int i = 0; i < cnt; i++)
    {
        int x = find(edge[i].u);
        int y = find(edge[i].v);
        //如果没有添加，就添加上去
        if (x == y)  continue;
        ans += edge[i].w;
        p[x] = y;
    }
}

int main()
{
    //freopen("D:\\txt.txt", "r", stdin);
    while (cin >> n)
    {
        for (int i = 1; i <= n; i++)
            p[i] = i;
        cnt = 0;
        for (int i = 1; i <= n;i++)
        for (int j = 1; j <= n; j++)
        {
            edge[cnt].u = i;
            edge[cnt].v = j;
            cin >> edge[cnt].w;
            cnt++;
        }
        int m;
        int a, b;
        cin >> m;
        while (m--)
        {
            cin >> a >> b;
            int x = find(a);
            int y = find(b);
            p[x] = y;
        }
        ans = 0;
        Kruskal();
        cout << ans << endl;
    }
    return 0;
}