接雨水 IIJava
文章发布较早，内容可能过时，阅读注意甄别。
# 题目
给你一个 m x n 的矩阵，其中的值均为非负整数，代表二维高度图每个单元的高度，请计算图中形状最多能接多少体积的雨水。
示例 1:
输入: heightMap = [[1,4,3,1,3,2],[3,2,1,3,2,4],[2,3,3,2,3,1]]
输出: 4
解释: 下雨后，雨水将会被上图蓝色的方块中。总的接雨水量为1+2+1=4。
示例 2:
输入: heightMap = [[3,3,3,3,3],[3,2,2,2,3],[3,2,1,2,3],[3,2,2,2,3],[3,3,3,3,3]]
输出: 10
提示:
- m == heightMap.length
- n == heightMap[i].length
- 1 <= m, n <= 200
- 0 <= heightMap[i][j] <= 2 * 104
# 思路
PriorityQueue
# 解法
class Solution {
public int trapRainWater(int[][] heightMap) {
int m=heightMap.length;
int n=heightMap[0].length;
//备忘录
boolean [][]visited=new boolean[m][n];
//根据高度构建最小堆
PriorityQueue<int[]> pq=new PriorityQueue<>((a1,a2)->{
return a1[2]-a2[2];
});
//先把最外围加入最小堆
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){
pq.offer(new int[]{i,j,heightMap[i][j]});
visited[i][j]=true;
}
}
}
int result=0;
//出队，往四周方向寻找更小的进行灌水
while (!pq.isEmpty()){
int[] poll = pq.poll();
int x = poll[0];
int y = poll[1];
int height = poll[2];
//左
if (x-1>=0&&!visited[x-1][y]){
if (heightMap[x-1][y]<height){
//可以灌水
result+=height-heightMap[x-1][y];
//灌水后的格子成为外围之一
pq.offer(new int[]{x-1,y,height});
}
else {
//没灌水也成为外围
pq.offer(new int[]{x - 1, y, heightMap[x - 1][y]});
}
visited[x-1][y]=true;
}
//右
if (x+1<m&&!visited[x+1][y]){
if (heightMap[x+1][y]<height){
//可以灌水
result+=height-heightMap[x+1][y];
//灌水后的格子成为外围之一
pq.offer(new int[]{x+1,y,height});
}else {
//没灌水也成为外围
pq.offer(new int[]{x + 1, y, heightMap[x + 1][y]});
}
visited[x+1][y]=true;
}
//下
if (y-1>=0&&!visited[x][y-1]){
if (heightMap[x][y-1]<height){
//可以灌水
result+=height-heightMap[x][y-1];
//灌水后的格子成为外围之一
pq.offer(new int[]{x,y-1,height});
}else {
//没灌水也成为外围
pq.offer(new int[]{x, y - 1, heightMap[x][y - 1]});
}
visited[x][y-1]=true;
}
//上
if (y+1<n&&!visited[x][y+1]){
if (heightMap[x][y+1]<height){
//可以灌水
result+=height-heightMap[x][y+1];
//灌水后的格子成为外围之一
pq.offer(new int[]{x,y+1,height});
}else {
//没灌水也成为外围
pq.offer(new int[]{x, y + 1, heightMap[x][y + 1]});
}
visited[x][y+1]=true;
}
}
return result;
}
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
# 总结
- 分析出几种情况，然后分别对各个情况实现
