【并查集】用水填坑

题目链接：https://ac.nowcoder.com/acm/contest/403/A

 先想想一维的降雨积水问题。很容易发现可以考虑每个地两侧的高度最大值的最小值，就是这个地方的积水量，只需要把这些积水量累和即可。但是二维的就没法这么做了。继续想想一维的问题，发现可以有一种并查集的解法，进而推广到二维。
 我们把所有的高度排序，从小到大添加入原图，用并查集维护连通块，如果新加入的值$a[i][j]$在之前某个连通块的一侧，那么说明这个连通块周围的“围墙”高度至少是$a[i][j]$，我们让答案增加上$连通块大小\ast (a[i][j]-连通块最大值)$。这连通块最大值我们可以通过把设置新加入点为根的方法，使得并查集的顶点就是最大值点。注意判断连通块是否和边界相连，以及连接的连通块是否重复的情况。这道题就解决了。

#include<cstdio>
#include<algorithm>
using namespace std;
using LL=long long;

const int dx[4]={1,0,-1,0};
const int dy[4]={0,1,0,-1};

int n,m,a[1000005],an,fa[1000005],sz[1000005];
bool edge[1000005];
pair<int,int> b[1000005];
LL ans;

inline int tr(int i, int j)
{
	return (i-1)*m+j;
}

inline void rtr(int &x, int &y, int d)
{
	x=(d-1)/m+1;
	y=(d-1)%m+1;
}

inline int fis(int x)
{
	return fa[x]==x?x:fa[x]=fis(fa[x]);
}

int main()
{
	scanf("%d%d",&n,&m);
	for(int i=1;i<=n;i++)
		for(int j=1;j<=m;j++)
			scanf("%d",&a[++an]),b[an]={a[an],an};
	sort(b+1,b+an+1);
	for(int i=1,u,v;i<=n*m;i++)
	{
		fa[b[i].second]=b[i].second;
		sz[b[i].second]=1;
		rtr(u,v,b[i].second);
		for(int j=0,p,q,h;j<4;j++)
		{
			p=u+dx[j],q=v+dy[j];
			h=tr(p,q);
			if(p>0&&p<=n&&q>0&&q<=m)
			{
				if(fis(h)&&fis(h)!=b[i].second)
				{
					if(!edge[fis(h)])
						ans+=(LL)(b[i].first-a[fis(h)])*sz[fis(h)];
					else
						edge[b[i].second]=true;
					sz[b[i].second]+=sz[fis(h)];
					fa[fis(h)]=b[i].second;
				}
			}
			else
				edge[b[i].second]=true;
		}
	}
	printf("%lld",ans);
	return 0;
}