题目如下:
给你一个 m x n 的矩阵,其中的值均为非负整数,代表二维高度图每个单元的高度,请计算图中形状最多能接多少体积的雨水。
输入: heightMap = [[1,4,3,1,3,2],[3,2,1,3,2,4],[2,3,3,2,3,1]]
输出: 4
解释: 下雨后,雨水将会被上图蓝色的方块中。总的接雨水量为1+2+1=4。
分析:
本题为经典题目,解题的原理和方法都可以参考「接雨水」,本题主要从一维数组变成了二维数组。
首先我们思考一下什么样的方块一定可以接住水:
该方块不为最外层的方块;
该方块自身的高度比其上下左右四个相邻的方块接水后的高度都要低;
我们假设方块的索引为 (i,j),方块的高度为heightMap[i][j],方块接水后的高度为water[i][j]。则我们知道方块 (i,j) 的接水后的高度为:
water[i][j]=max(heightMap[i][j],min(water[i−1][j],water[i+1][j],water[i][j−1],water[i][j+1]))
我们知道方块 (i,j)(i,j) 实际接水的容量计算公式为water[i][j]−heightMap[i][j]。
首先我们可以确定的是,矩阵的最外层的方块接水后的高度就是方块的自身高度,因为最外层的方块无法接水,因此最外层的方块 water[i][j]=heightMap[i][j]。
根据短板原理,接到的雨水的高度由这个容器周围最短的木板来确定的。我们可以知道容器内水的高度取决于最外层高度最低的方块。
我们假设已经知道最外层的方块接水后的高度的最小值,则此时我们根据木桶原理,肯定可以确定最小高度方块的相邻方块的接水高度。我们同时更新最外层的方块标记,我们在新的最外层的方块再次找到接水后的高度的最小值,同时确定与其相邻的方块的接水高度,
然后再次更新最外层,依次迭代直到求出所有的方块的接水高度,即可知道矩阵中的接水容量。
代码如下:
class Solution {
public int trapRainWater(int[][] heightMap) {
if (heightMap.length <= 2 || heightMap[0].length <= 2) {
return 0;
}
int m = heightMap.length;
int n = heightMap[0].length;
// 访问过的节点
boolean[][] visit = new boolean[m][n];
// 小顶堆,且放入的是数组,比较高度
PriorityQueue<int[]> pq = new PriorityQueue<>((o1, o2) -> o1[1] - o2[1]);
for (int i = 0; i < m; ++i) {
for (int j = 0; j < n; ++j) {
if (i == 0 || i == m - 1 || j == 0 || j == n - 1) {
// i * n + j代表第几个
// 0 1 2 //4*3d
// 3 4 5
// 6 7 8
// 9 10 11
pq.offer(new int[]{i * n + j, heightMap[i][j]});
visit[i][j] = true;
}
}
}
int res = 0;
// 通过for循环4次则能用i和i+1定位【上,右,下,左】
int[] dirs = {-1, 0, 1, 0, -1};
// 如果小顶堆为空
while (!pq.isEmpty()) {
// 小顶堆出栈
int[] curr = pq.poll();
for (int k = 0; k < 4; ++k) {
// 第几个取位置,第/n则是行数index,第%n则是第几列数index
int nx = curr[0] / n + dirs[k];
int ny = curr[0] % n + dirs[k + 1];
// 不越界并且没有被访问过
if (nx >= 0 && nx < m && ny >= 0 && ny < n && !visit[nx][ny]) {
// 如果当前位置大于周围,则可以注水,并且访问过之后高度为注水后的高度 ※因为这是从最小开始的,比最小还小的则可以接雨水
if (curr[1] > heightMap[nx][ny]) {
res += curr[1] - heightMap[nx][ny];
pq.offer(new int[]{nx * n + ny, curr[1]});
} else { // 如果大于则更新边界
pq.offer(new int[]{nx * n + ny,heightMap[nx][ny]});
}
// 将访问过的位置设置为true
visit[nx][ny] = true;
}
}
}
return res;
}
}