还是最小生成树

老 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
#include<stdio.h>
#include<string.h>
#include<algorithm>
#include<math.h>
#define INF 0x3f3f3f3f
#define I 2000010
using namespace std;
int map[1000][1001];
int N,M,E,pre[I];

struct edge
{
	int u,v,cost;
};
edge e[I];

bool cmp(edge e1,edge e2)
{
	return e1.cost<e2.cost;
}

void init()
{
	for(int i=0;i<=M*N;i++)
		pre[i]=i;
}

int Find (int x)
{
	if(x==pre[x]) return x;
	return pre[x]=Find(pre[x]);
}

int kruskal()
{
	int ans=0,cnt=0;
	sort(e,e+E,cmp);
	init();
	for(int i=0;i<E;i++)
	{
		int fx=Find(e[i].u);
		int fy=Find(e[i].v);
		if(fx!=fy)
		{
			pre[fx]=fy;
			ans+=e[i].cost;
			cnt++;
		}
		if(cnt==M*N-1)
			break;
	}
	return ans;
}

int main()
{
	int cas=1;
	int T;scanf("%d",&T);
	while(T--)
	{
		memset(e,0,sizeof(e));
		memset(map,0,sizeof(map));
		scanf("%d%d",&N,&M);
		for(int i=0;i<N;i++)
			for(int j=0;j<M;j++)
				scanf("%d",&map[i][j]);
		int cnt=0;
		for(int i=0;i<N;i++)
		{
			for(int j=0;j<M;j++)
			{
				if(j!=M-1)
				{
					e[cnt].u=i*M+j;
					e[cnt].v=i*M+j+1;
					e[cnt++].cost=fabs(map[i][j]-map[i][j+1]);
				}
				if(i!=N-1)
				{
					e[cnt].u=i*M+j;
					e[cnt].v=i*M+j+M;
					e[cnt++].cost=fabs(map[i][j]-map[i+1][j]);
				}
			}
		}
		E=cnt+1;
		int t=kruskal();
	printf("Case #%d:\n%d\n",cas++,t);
	}
	return 0;
}

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值