HDU 5253（最小生成树

这题比赛的时候写挂了，WA了将近十次也没找到问题。。。。刚刚重新写了一次，mle了一次。。这题卡内存还是挺紧的，然后改了下就a了。。。比赛的时候简直背啊。。。

#include<iostream>
#include<map>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cstdlib>
#include<vector>
#define pb push_back
using namespace std;
typedef long long ll;
const int maxv=1e6+40;
int T,cas;
int N,M;
struct edge{
    int u,v,cap;
};
vector<edge> G;
int field[1005][1005];
int fa[maxv];
void initFa(){
    for(int i=0;i<=N*M;i++) fa[i]=i;
}
int findFa(int x){
    if(fa[x]==x) return x;
    else return fa[x]=findFa(fa[x]);
}
bool same(int x,int y){
    return findFa(x)==findFa(y);
}
void unite(int x,int y){
    fa[findFa(x)]=findFa(y);
}
int kruskall(){
    int added=0,ans=0,h=0;
    initFa();
    while(added<N*M-1){
        int x=G[h].u,y=G[h].v,c=G[h].cap;
        h++;
        if(!same(x,y)) ans+=c,added++,unite(x,y);
    }
    return ans;
}
int drx[]={0,1};
int dry[]={1,0};
bool inRange(int x,int y){
    return x>=0&&x<N&&y>=0&&y<M;
}
int getN(int x,int y){
    return x*M+y;
}
void constuct(){
    G.clear();
    for(int i=0;i<N;i++){
        for(int j=0;j<M;j++){
            for(int k=0;k<2;k++){
                int dx=drx[k]+i,dy=dry[k]+j;
                if(inRange(dx,dy)){
                    G.pb((edge){getN(i,j),getN(dx,dy),abs(field[i][j]-field[dx][dy])});
                }
            }
        }
    }
}
bool cmp(edge a,edge b){
    return a.cap<b.cap;
}
int main(){
    cin>>T;
    while(T--){
        cin>>N>>M;
        for(int i=0;i<N;i++){
            for(int j=0;j<M;j++){
                scanf("%d",&field[i][j]);
            }
        }
        constuct();
        sort(G.begin(),G.end(),cmp);
        printf("Case #%d:
%d
",++cas,kruskall());
    }
    return 0;
}