POJ 2421 Constructing Roads（最小生成树）

一开始理解题意有点困难，给出n*n的一个矩阵，其中（i，j）这个坐标的值表示从i村 到 j村 的距离，不过有一些村庄已经被修好了，所以把他们加入到并查集中去就行了，被修好的路不需要花费费用，求最小费用，典型的最小生成树做法，这次用kruskal做法，但是这题用到两个不同的合并函数，各有用处

#include<iostream>
#include<cstring>
#include<cstdio>
#include<algorithm>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<cmath>
using namespace std;
int a[105][105];
int pre[105*105];
struct node
{
    int u;
    int v;
    int len;
}e[105*105];
int find(int a)
{
    if(pre[a] == a)return a;
    else return find(pre[a]);
}
void join(int a, int b)//合并已经修好的道路进入并查集
{
    int i = find(a);
    int j = find(b);
    if(i != j)pre[i] = j;
}
int join1(int a, int b)//合并未修好的道路进入并查集
{
    int i = find(a);
    int j = find(b);
    if(i != j)
    {
        pre[i] = j;
        return 1;
    }
    return 0;
}
bool cmp(node a, node b)
{
    return a.len < b.len;
}
int main()
{
    int n;
    while(scanf("%d",&n)!=EOF)
    {
        int cnt = 1;
        int cnt1 = 1;
        for(int i = 1; i <= n; i ++)
            for(int j = 1; j <= n; j ++)
        {
            scanf("%d",&a[i][j]);
            pre[cnt++] = (i - 1) * n + j;
            e[cnt1].u = i;
            e[cnt1].v = j;
            e[cnt1++].len = a[i][j];
        }
//        for(int i = 1; i < cnt ; i ++)
//            cout<<pre[i]<<' ';
        int q, u, v;
        long long ans = 0;
        scanf("%d",&q);
        while(q--)
        {
            scanf("%d%d",&u, &v);
            join(u, v);
        }
//        for(int i = 1; i < cnt1; i ++)
//            cout<<e[i].u<<' '<<e[i].v<<' '<<e[i].len<<endl;
        sort(e + 1, e + cnt1, cmp);
//        for(int i = 1; i < cnt1; i ++)
//            cout<<e[i].u<<' '<<e[i].v<<' '<<e[i].len<<endl;
        for(int i = 1; i < cnt1; i ++)
        {
            if(join1(e[i].u, e[i].v))
                ans += e[i].len;
        }
        printf("%lld\n",ans);
    }
    return 0;
}

另外需要注意边界问题，一开始没有把控好边界wa了两次