HDU 5253 连接的管道(最小生成树-Kruskal+并查集)

连接的管道

Time Limit: 2000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 2016 Accepted Submission(s): 709

Problem Description

老 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
  
  

Source

2015年百度之星程序设计大赛 - 初赛(2) 

/*
    思路：一共1000*1000*2条边 
*/
#include<iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int N=1000+10;
struct Node{
	int s,end,w;
}node[N*N*2];
bool cmp(Node a,Node b){
	return a.w<b.w;
}
int pre[N*N]; //点i的父节点 
int n,m;   //矩阵长宽 
int len;   //边数目 
int map[N][N];   
int step[2][2]={{0,1},{1,0}};  //右和下 
inline void Init(){
	for(int i=0;i<=n*m;i++)
	  pre[i]=i;
} 
inline int find(int x){
	int r=x;
	while(pre[r]!=r)   //找父节点
	   r=pre[r];
	//现在r是父节点 
	int i=x,j; 
	while(i!=r){   //路径压缩 
		j=pre[i];
		pre[i]=r;
		i=j;
	} 
	return r; 
}
inline bool join(int x,int y){
	int fx=find(x);
	int fy=find(y);
	if(fx!=fy){  //不存在回路 
		pre[fx]=fy;
		return true;
	}
	return false;
}
inline int kruskal(){
	int i,count=1,sum=0;
	Init();
	sort(node,node+len,cmp);
	for(i=0;count<n*m;i++){    //最小生成树n个点，就总共n-1条边
	//这里n*m个点，就n*m-1条边 
		if(join(node[i].s,node[i].end)){
			sum+=node[i].w;
			count++;
		}
	}
	return sum;  //最小花费树 
}
int main() 
{ 
    int t,re,now;
	scanf("%d",&t);
	for(now=1;now<=t;now++){
		re=0;
		len=0;
		scanf("%d%d",&n,&m);
		for(int i=0;i<n;i++)
		for(int j=0;j<m;j++)
		   scanf("%d",&map[i][j]);
		for(int i=0;i<n;i++)    //枚举边 对于每个点，枚举和右边连边和下面的连边 
		for(int j=0;j<m;j++)
		{
			for(int k=0;k<2;k++){
				int a=i+step[k][0];
				int b=j+step[k][1];
				if(b>=m||a>=n)
				  continue;
				Node tmp;
				tmp.s=i*m+j;  //起点 
				tmp.end=a*m+b;    //终点 
			    tmp.w=abs(map[a][b]-map[i][j]);
			    node[len++]=tmp;  
			}			
		}
		re=kruskal();
		printf("Case #%d:\n",now);
		printf("%d\n",re); 
	} 	
    return 0;
}