题目地址:
https://www.acwing.com/problem/content/594/
海里有一个小岛。岛可以描述为一个具有 R R R行 C C C列的矩阵,其中包含的数字 H [ i ] [ j ] H[i][j] H[i][j]表示每个单元格的高度。以下是一个 3 × 3 3×3 3×3岛屿的示例:
3 5 5
5 4 5
5 5 5
有时,大雨会在这个岛上的每个单元格上均匀地降落。你可以假设降水量可以是无限大。在这样的大雨之后,岛上的一些区域(由沿着边缘连接的一个或多个单元格形成)可能会存在积水。一片区域可以存在积水的前提是与该区域内单元格沿着边缘连接的区域外的单元格的高度均高于区域内单元格。周围的海洋被视为高度为 0 0 0的无限单元格。否则,水将始终流入其他区域并最终出海。你可以假设海的高度永远不会改变。在大雨过后,我们将使用 W [ i ] [ j ] W[i][j] W[i][j]来表示岛屿上每个单元格的高度。以下是对大雨过后的示例岛屿的高度的分析。最中央的初始高度为 4 4 4的单元格被初始高度为 5 5 5的单元格包围着,因此在该单元格将产生积水,在积水高度到达 5 5 5以后,没有更多区域被高度更高的单元格包围,因此将不再继续积水。请注意,水不会直接在两个共享顶点的单元格之间流动;水必须沿着共同的边缘流动,所以中央单元格的积水不会从左上角处流走。以下是雨后示例岛屿的高度:
3 5 5
5 5 5
5 5 5
给定岛的高度矩阵,你能计算出大雨后的每个单元格的增加高度 ( W [ i ] [ j ] − H [ i ] [ j ] ) (W[i][j]−H[i][j]) (W[i][j]−H[i][j])的总和吗?
输入格式:
第一行包含整数
T
T
T,表示共有
T
T
T组测试数据。每组数据第一行包含两个整数
R
R
R和
C
C
C。接下来
R
R
R行每行包含
C
C
C个整数
H
[
i
]
[
j
]
H[i][j]
H[i][j],表示矩阵内单元格的高度。
输出格式:
每组数据输出一个结果,每个结果占一行。结果表示为Case #x: y
,其中
x
x
x是组别编号(从
1
1
1开始),
y
y
y是增加高度的总和。
数据范围:
1
≤
T
≤
100
1≤T≤100
1≤T≤100
1
≤
H
[
i
]
[
j
]
≤
1000
1≤H[i][j]≤1000
1≤H[i][j]≤1000
1
≤
R
,
C
≤
50
1≤R,C≤50
1≤R,C≤50
每个位置的水量是由其外围的最矮的柱子的高度减去其自己的柱子高度决定的(当然如果减出来是负数则视为 0 0 0),所以可以用最小堆,先将最外围的柱子入堆,然后每出堆一个柱子,就看一下其四周哪个柱子的水量没算出过,累加其水量后,再将注水后的高度和坐标入堆(好像注水之后柱子长高了一样)。参考https://blog.csdn.net/qq_46105170/article/details/108575830。代码如下:
#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
// 第一维存柱子高度,后面两维存坐标
typedef pair<int, pair<int, int> > PII;
const int R = 55, C = 55, d[] = {1, 0, -1, 0, 1};
int r, c;
int h[R][C];
bool st[R][C];
int get_vol() {
memset(st, 0, sizeof st);
priority_queue<PII, vector<PII>, greater<PII> > heap;
for (int i = 1; i <= r; i++) {
heap.push({h[i][1], {i, 1}}), heap.push({h[i][c], {i, c}});
st[i][1] = st[i][c] = true;
}
for (int i = 1; i <= c; i++) {
heap.push({h[1][i], {1, i}}), heap.push({h[r][i], {r, i}});
st[1][i] = st[r][i] = true;
}
int res = 0;
while (heap.size()) {
auto t = heap.top(); heap.pop();
int hei = t.first, x = t.second.first, y = t.second.second;
for (int i = 0; i < 4; i++) {
int nx = x + d[i], ny = y + d[i + 1];
if (!st[nx][ny] && 1 <= nx && nx <= r && 1 <= ny && ny <= c) {
res += max(0, hei - h[nx][ny]);
heap.push({max(hei, h[nx][ny]), {nx, ny}});
// 算出(nx, ny)这个位置的水量之后就标记一下
st[nx][ny] = true;
}
}
}
return res;
}
int main() {
int T;
cin >> T;
for (int t = 1; t <= T; t++) {
cin >> r >> c;
for (int i = 1; i <= r; i++)
for (int j = 1; j <= c; j++)
cin >> h[i][j];
printf("Case #%d: %d\n", t, get_vol());
}
return 0;
}
时间复杂度 O ( R C log ( R + C ) ) O(RC\log (R+C)) O(RClog(R+C)),空间 O ( R C ) O(RC) O(RC)。