hdu 5253连接的管道

传送门：http://acm.hdu.edu.cn/showproblem.php?pid=5253
题意：农夫连接农田，每块农田的相对高度差是建图的权值。

最小生成树的入门模版题，不错不错

#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;//操作完毕，连通两块农田
        }
    }
    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];
                    int ny=j+dir[d][1];
                    //右下，右下循环
                    //（这个循环very nice!）
                    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]);  
                    }
                }
        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);
    }
}