【集训DAY8】water【最小生成树】

思路：

我们考虑一个块的高度就是从这个块走出矩形的所有路径上的最 大值的最小值。

相邻块连边，权值为两块的较大值，矩形 边界的块向“矩形外”连边，权值为max(高度，0)，做最 小生成树。

$code$

#include<iostream>
#include<cstdio>
#include<algorithm>

using namespace std;

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

int n, m, a[305][305], fa[100010];
int head[100010], tot, tot2, h[100010], ans[100010];
struct node {
	int from, to, next, w;
}b[4000010], e[100010];

int dian(int i, int j) {
	if(i < 1 || i > n || j < 1 || j > m) return 0;
	return (i - 1) * m + j;
}

void add(int x, int y, int w) {
	b[++ tot] = (node) { x, y, 0, w };
}

bool cmp(node x, node y) {
	return x.w < y.w;
} 

int getfa(int x) {
	if(x == fa[x]) return x;
	return fa[x] = getfa(fa[x]);
}

void add2(int x, int y, int w) {
	e[++ tot2] = (node) { x, y, head[x], w };
	head[x] = tot2;
}

void krus() {
	sort(b + 1, b + 1 + tot, cmp);
	for(int i = 0; i <= n * m; i ++) fa[i] = i;
	for(int i = 1; i <= tot; i ++) {
		int x = b[i].from, y = b[i].to;
		int xx = getfa(x), yy = getfa(y);
		if(xx != yy) fa[xx] = yy, add2(x, y, b[i].w), add2(y, x, b[i].w);
	}
}

void dfs(int x, int fa) {
	for(int i = head[x]; i; i = e[i].next) {
		int y = e[i].to;
		if(y == fa) continue;
		ans[y] = max(ans[x], e[i].w);
		dfs(y, 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[i][j]), h[dian(i, j)] = a[i][j];
	for(int i = 1; i <= n; i ++)
		for(int j = 1; j <= m; j ++) {
			for(int k = 0; k < 4; k ++)
				add(dian(i, j), dian(i + dx[k], j + dy[k]), max(a[i][j], a[i + dx[k]][j + dy[k]]));
		}
	krus();
	dfs(0, 0);
	for(int i = 1; i <= n * m; i ++) {
		printf("%d ", ans[i] - h[i]);
		if(i % m == 0) printf("\n");
	}
	return 0;
}