洛谷 P4807 抽水机

题目背景

kkk被Farmer John和他的奶牛贝茜虐的很惨，然后她也想体验下一个Farmer的生活。但她又懒得种地，就选择养鱼。

题目描述

这些鱼都是热带鱼（废话），很娇贵（比kkk娇贵），要经常换水，要不然每当kkk走过来的时候鱼们就会一起使劲拍尾巴导致kkk并不情愿的洗个冷水澡（别问我热带鱼为毛这么机智）。但kkk并不勤快，他只想花费最少的力气以实现换水。

kkk的鱼塘可以分成n*n个独立小池，每两个相邻的小池间都有一个水闸控制水位。开启一个水闸需要花费的力气是这两个相邻的小池的水位之差。已知各个小池的水位，kkk想知道她要给每个小池都换水至少需要多少力气。

输入输出格式

输入格式：

第一行一个整数n

接下来n*n个数表示各个小池的水位

输出格式：

最小力气

输入输出样例

输入样例#1：

3
1 2 3
4 5 6
7 8 9

输出样例#1：

12

说明

1<=n<=100

1<=水位<=100

~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

最小生成树~

加边比较麻烦，我是按从左往右从上往下编号的，但实际上如果重边也可以，但是循环的次数会多些。

玄学问题6号：abs函数不能用define，只能写成函数……迷……

（第一次提交RE了，然后好好的100就变成了95！！！所以说最小生成树存边数组一定要开大啊啊啊啊！！！）

#include<cstdio>
#include<algorithm>
using namespace std;

int n,a[1001][1001],cnt,ans,fa[100001];

struct node{
	int x,y,v;
}ro[1000001];

int abs(int u)
{
return u>0 ? u:-u;
}

bool cmp(node u,node v)
{
	return u.v<v.v;
}

int findd(int u)
{
	return fa[u]==u ? u:fa[u]=findd(fa[u]);
}

int main()
{
	scanf("%d",&n);
	for(int i=1;i<=n;i++)
	  for(int j=1;j<=n;j++) scanf("%d",&a[i][j]),fa[i*n-n+j]=i*n-n+j;
	for(int i=1;i<n;i++)
	  for(int j=1;j<n;j++)
	  {
	  	ro[++cnt].v=abs(a[i][j]-a[i][j+1]);ro[cnt].x=i*n-n+j,ro[cnt].y=i*n-n+j+1;
	  	ro[++cnt].v=abs(a[i][j]-a[i+1][j]);ro[cnt].x=i*n-n+j,ro[cnt].y=i*n+j;
	  }
	for(int i=2;i<=n;i++)
	{
		ro[++cnt].v=abs(a[i][n]-a[i-1][n]);ro[cnt].x=i*n,ro[cnt].y=i*n-n;
	  	ro[++cnt].v=abs(a[n][i]-a[n][i-1]);ro[cnt].x=n*n-n+i,ro[cnt].y=n*n-n+i-1;
	}
	sort(ro+1,ro+cnt+1,cmp);
	for(int i=1;i<=cnt;i++)
	  if(findd(ro[i].x)!=findd(ro[i].y))
	  {
	  	fa[findd(ro[i].x)]=findd(ro[i].y);ans+=ro[i].v;
	  }
	printf("%d\n",ans);
	return 0;
}