HDU5253(最小生成树)

//
//最小生成树有很多种算法   习惯用并查集+克鲁斯卡尔
//  main.c
//  example
//
//  Created by Adam on 15/2/2.
//  Copyright (c) 2015年 Adam. All rights reserved.
//

#include <iostream>
#include "stdio.h"
#include "stdlib.h"
#include "string.h"
#include "algorithm"
#include <queue>
using namespace std;

int ss[1005][1005];
int father[800000];

int n,m;
struct node {
int op,ed;
int ans;
}edge[2000005];

int find(int x)
{
if(x==father[x]) return x;
else return father[x]=find(father[x]);
}

int cmp(node x,node y)
{
return x.ans<y.ans;
}

int main()
{
int t,pos,len,ans,cnt=0;
scanf("%d",&t);
while(t--)
{
pos=0;
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
scanf("%d",&ss[i][j]);

for(int i=0;i<n;i++)
for(int j=0;j<m;j++)
{
if(j!=m-1){
edge[pos].op=i*m+j;
edge[pos].ed=i*m+j+1;
edge[pos++].ans=abs(ss[i][j]-ss[i][j+1]);
}
if(i!=n-1){
edge[pos].op=i*m+j;
edge[pos].ed=i*m+m+j;
edge[pos++].ans=abs(ss[i][j]-ss[i+1][j]);
}
}

for(int i=0;i<n*m;i++)
{
father[i]=i;
}
sort(edge,edge+pos,cmp);
len=0;ans=0;
for(int i=0;i<pos;i++)
{
int a=edge[i].op;
int b=edge[i].ed;
a=find(a);
b=find(b);
if(a==b) continue;
father[a]=b;
len++;
ans+=edge[i].ans;
if(len==(n*m-1)) break;
}
printf("Case #%d:\n%d\n",++cnt,ans);
}
return 0;
}