HDU 5253 连接的管道

就是就最小生成树，就是有点坑人

#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
using namespace std;
#define N 1005
int a[N][N];
int p[N*N];
int m,n;
int L;
struct node
{
    int u,v,c;
}t;
struct node M[N*N*4];//这样更节约内存
int cmp(node p,node q)
{
    return p.c<q.c;
}
int find(int k)
{
    if(p[k]!=k)
    {
        return p[k]=find(p[k]);
    }
    else
    {
        return p[k];
    }

}
int kus()
{
    int i,sum=0,cnt=0;
    for(i=0; i<L; i++)
    {
        int x=find(M[i].u);
        int y=find(M[i].v);
        if(x!=y)
        {
            p[y]=x;
            sum+=M[i].c;
            cnt++;
        }
        if(cnt==m*n-1)
            return sum;
    }
    return sum;
}
int main()
{
    int T;
    scanf("%d",&T);
    int k=1;
    while(T--)
    {
        scanf("%d%d",&m,&n);
        int i,j;
        L=0;
        for(i=0; i<m; i++)
        {
            for(j=0; j<n; j++)
            {
                scanf("%d",&a[i][j]);
                p[i*n+j]=i*n+j;
                if(i>0)
                {
                    t.u=i*n+j;
                    t.v=i*n+j-n;
                    t.c=abs(a[i][j]-a[i-1][j]);
                    M[L++]=t;
                }
                if(j>0)
                {
                    t.u=i*n+j;
                    t.v=i*n+j-1;
                    t.c=abs(a[i][j]-a[i][j-1]);
                    M[L++]=t;
                }
            }
        }
        sort(M,M+L,cmp);
        int sum=kus();
        printf("Case #%d:\n",k++);
        printf("%d\n",sum);
    }
    return 0;
}