标签：优先级 math 一个 static 二维 while || void src
给定一个 m x n 的矩阵，其中的值均为正整数，代表二维高度图每个单元的高度，请计算图中形状最多能接多少体积的雨水。
说明:
m 和 n 都是小于110的整数。每一个单位的高度都大于0 且小于 20000。
示例：
给出如下 3x6 的高度图:
[
[1,4,3,1,3,2],
[3,2,1,3,2,4],
[2,3,3,2,3,1]
]
返回 4。
如上图所示，这是下雨前的高度图[[1,4,3,1,3,2],[3,2,1,3,2,4],[2,3,3,2,3,1]] 的状态。
下雨后，雨水将会被存储在这些方块中。总的接雨水量是4。
三维的装水的问题，主要思路还是从边缘的地方开始找。
priorityqueue每次poll出最高优先级的元素，也就是目前height最底的元素。
对于visit之后的元素来说，新的高度是要updata的，要么保持原样，要么变成装了水之后的高度。
1 import java.util.Comparator; 2 import java.util.PriorityQueue; 3 4 public class Solution { 5 6 private static class Cell { 7 private int row; 8 private int col; 9 private int height; 10 11 public Cell(int row, int col, int height){ 12 this.row = row; 13 this.col = col; 14 this.height = height; 15 } 16 } 17 18 19 public static int trapRainWater(int[][] heightMap) { 20 if(heightMap == null || heightMap.length == 0 ||heightMap[0].length == 0) { 21 return 0; 22 } 23 24 PriorityQueue<Cell> queue = new PriorityQueue<>(1, new Comparator<Cell>(){ 25 public int compare(Cell a, Cell b) { 26 return a.height - b.height; 27 } 28 }); 29 30 int m = heightMap.length; 31 int n = heightMap[0].length; 32 boolean[][] visited = new boolean[m][n]; 33 34 for (int i = 0; i< m ;i++){ 35 visited[i][0] = true; 36 visited[i][n-1] = true; 37 queue.offer(new Cell(i, 0, heightMap[i][0])); 38 queue.offer(new Cell(i, n-1, heightMap[i][n-1])); 39 } 40 41 for (int i = 0; i< n ;i++){ 42 visited[0][i] = true; 43 visited[m-1][i] = true; 44 queue.offer(new Cell(0, i, heightMap[0][i])); 45 queue.offer(new Cell(m-1, i, heightMap[m-1][i])); 46 } 47 48 49 50 int[][] dirs = new int[][]{{-1,0},{1,0},{0,-1},{0,1}}; 51 int res = 0; 52 while(!queue.isEmpty()){ 53 Cell cell = queue.poll(); 54 for(int[] dir : dirs){ 55 int row = cell.row + dir[0]; 56 int col = cell.col + dir[1]; 57 if(row >= 0 && row < m && col >=0 && col < n && !visited[row][col]){ 58 visited[row][col] = true; 59 res += Math.max(0,cell.height - heightMap[row][col]); 60 queue.offer(new Cell(row,col,Math.max(heightMap[row][col],cell.height))); 61 } 62 } 63 } 64 return res; 65 } 66 67 public static void main(String[] args){ 68 int[][] heightMap={{1,4,3,1,3,2},{3,2,1,3,2,4},{2,3,3,2,3,1}}; 69 trapRainWater(heightMap); 70 } 71 }
int[][] heightMap={{1,4,3,1,3,2},{3,2,1,3,2,4},{2,3,3,2,3,1}};
trapRainWater(heightMap); }}
标签：优先级 math 一个 static 二维 while || void src
原文地址：https://www.cnblogs.com/kexinxin/p/10241839.html
