HDU 5253 连接的管道

题目链接：连接的管道

题面：

连接的管道

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

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)

解题：

建一张图，求最小生成树。不过貌似搜索的方法要好一些，算是水过吧，没能充分利用一个点只与周围四点相连这个条件。

代码：

#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
int map[1005][1005];
struct set
{
	int parent;  
	int rank;    
}elem[1000000+10];
void init(int x)
{
	int i;
	for(i=1;i<=x;i++)
	{
		elem[i].parent=i;
		elem[i].rank=1;
	}
}
struct edge
{
	int a,b,val;
}storage[2000000+10];
bool cmp(edge a,edge b)
{
	return a.val<b.val; 
}
int Find(int x)
{
	int root,temp;
	temp=x;
	while(x!=elem[x].parent)    //寻找根节点
		x=elem[x].parent;
	root=x;
	x=temp;
	while (x!=elem[x].parent)   //压缩路径，全部赋值为根节点的值
	{
		temp=elem[x].parent;
		elem[x].parent=root;
		x=temp;
	}
	return root;
}
void Union(int a,int b)   //合并两个集合
{
	int x,y;
	x=Find(a);
	y=Find(b);
	if(elem[x].rank>=elem[y].rank)
	{
		elem[y].parent=elem[x].parent;
		elem[x].rank+=elem[y].rank;
	}
	else
	{
		elem[x].parent=elem[y].parent;
		elem[y].rank+=elem[x].rank;
	}
}
int main()
{
	int t,n,m,cnt,total,num,ans,ccnt,aa,bb;
	scanf("%d",&t);
	for(int i=1;i<=t;i++)
	{
		printf("Case #%d:\n",i);
		cnt=ccnt=ans=0;
		scanf("%d%d",&n,&m);
		//total=(n*m*4-4*2-(n-2)-(m-2))/2;总边数 
		num=n*m;
		init(num);
		for(int j=1;j<=n;j++)
		{
			for(int k=1;k<=m;k++)
			scanf("%d",&map[j][k]);
		} 
		for(int j=1;j<n;j++)
		{
			int hang=(j-1)*m;
			for(int k=1;k<m;k++)
			{
				storage[cnt].a=hang+k;
				storage[cnt].b=hang+k+1;
				storage[cnt++].val=abs(map[j][k]-map[j][k+1]);
				storage[cnt].a=hang+k;
				storage[cnt].b=hang+m+k;
				storage[cnt++].val=abs(map[j][k]-map[j+1][k]);
			}
			storage[cnt].a=hang+m;
			storage[cnt].b=hang+m+m;
			storage[cnt++].val=abs(map[j][m]-map[j+1][m]);
		}
		//cout<<"cnt: "<<cnt<<endl;
		for(int j=1;j<m;j++)
		{
			storage[cnt].a=(n-1)*m+j;
			storage[cnt].b=(n-1)*m+j+1;
			storage[cnt++].val=abs(map[n][j]-map[n][j+1]);
		}
		sort(storage,storage+cnt,cmp);
		//cout<<"Edge: "<<cnt<<" "<<total<<endl;
		for(int j=0;j<cnt;j++)
		{
			aa=storage[j].a;
			bb=storage[j].b;
			int t1=Find(aa);
			int t2=Find(bb);
			if(t1!=t2)
			{
				Union(aa,bb);
				ans+=storage[j].val;
				ccnt++;
			}
			if(ccnt==num-1)
			break;
		}
		printf("%d\n",ans);
	}
	return 0;
}