HDU - 5253 连接的管道

最小生成树的入门题：
最小生成树有两个算法，一个是prim算法，一个是krusal算法。最小生成树的前提是图是连通的。(算法详解，等深度学习后再来补)
老 Jack 有一片农田，以往几年都是靠天吃饭的。但是今年老天格外的不开眼，大旱。所以老 Jack 决定用管道将他的所有相邻的农田全部都串联起来，这样他就可以从远处引水过来进行灌溉了。当老 Jack 买完所有铺设在每块农田内部的管道的时候，老 Jack 遇到了新的难题，因为每一块农田的地势高度都不同，所以要想将两块农田的管道链接，老 Jack 就需要额外再购进跟这两块农田高度差相等长度的管道。

现在给出老 Jack农田的数据，你需要告诉老 Jack 在保证所有农田全部可连通灌溉的情况下，最少还需要再购进多长的管道。另外，每块农田都是方形等大的，一块农田只能跟它上下左右四块相邻的农田相连通。

Input
第一行输入一个数字T(T≤10)，代表输入的样例组数 

输入包含若干组测试数据，处理到文件结束。每组测试数据占若干行，第一行两个正整数 N,M(1≤N,M≤1000)，代表老 Jack 有N行*M列个农田。接下来 N 行，每行 M 个数字，代表每块农田的高度，农田的高度不会超过100。数字之间用空格分隔。 
Output
对于每组测试数据输出两行： 

第一行输出："Case #i:"。i代表第i组测试数据。 

第二行输出 1 个正整数，代表老 Jack 额外最少购进管道的长度。
Sample Input
2
4  3
9 12 4
7 8 56
32 32 43
21 12 12
2  3
34 56 56
12 23 4
Sample Output
Case #1:
82
Case #2:
74

中文题意很清楚，权值就是农田的相差高度。

#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
const int maxn =1010;
int fa[5000005];
int maze[maxn][maxn];
int dir[2][2]={{0,1},{1,0}};
int n,m;
int cnt,ans;
struct node{
    int u;
    int v;
    int w;
}edge[maxn*maxn*2];//这里要开两倍大，因为只记录每个农田的右、下两步，
//故边数大小不超过两倍的农田数。
bool cmp(node a,node b){
    return  a.w< b.w;
}
int find(int x){
    if(fa[x]==x)return x;
    else return fa[x]=find(fa[x]);
}
int kru(){
    ans=0;
    //printf("cnt=%d",cnt);
    for(int i=0;i<cnt;i++){
        int f1=find(edge[i].u);
        int f2=find(edge[i].v);
        if(f1!=f2){//查看是不是处在一个集合中，就看他们的根节点是不是同一个
            ans+=edge[i].w;//不在同一个根节点，说明两块农田之间没有相互连通，加上最小权值
            fa[f1]=f2;//操作完毕，连通两块农田；
        }
    }
    //printf("1\n");
    //printf("%d\n",ans);
    return ans;
}
int main(){
    int t;
    int cas=1;
    scanf("%d",&t);
    while(t--){
        cnt=0;
        scanf("%d%d",&n,&m);
        int k=n*m;
        for(int i=0;i<k;i++)
            fa[i]=i;
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                scanf("%d",&maze[i][j]);
        for(int i=0;i<n;i++)
            for(int j=0;j<m;j++)
                for(int d=0;d<2;d++)
                {
                    int nx=i+dir[d][0];//往右 //d[0][0]=0,d[0][1]=1;
                    int ny=j+dir[d][1];//往下  d[1][0]=1,d[1][1]=1;
                    if(nx>=0&&nx<n&&ny>=0&&ny<m){
                        edge[cnt].u=i*m+j;
                        edge[cnt].v=nx*m+ny;
                        edge[cnt++].w=abs(maze[i][j]-maze[nx][ny]);
                        //cnt记录边数
                    }
                }
        sort(edge,edge+cnt,cmp);
        //for(int i=0;i<cnt;i++)
            //printf("%d %d %d\n",edge[i].u,edge[i].v,edge[i].w);
        int tot=kru();
        printf("Case #%d:\n",cas++);
        printf("%d\n",tot);
    }
}