Leetcode 407.接雨水

接雨水

给定一个 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的，要么保持原样，要么变成装了水之后的高度。

 import java.util.Comparator;
 import java.util.PriorityQueue;

 public class Solution {

     private static class Cell {
         private int row;
         private int col;
         private int height;

         public Cell(int row, int col, int height){
             this.row = row;
             this.col = col;
             this.height = height;
         }
     }

     public static int trapRainWater(int[][] heightMap) {
         if(heightMap == null || heightMap.length == 0 ||heightMap[0].length == 0) {
             return 0;
         }

         PriorityQueue<Cell> queue = new PriorityQueue<>(1, new Comparator<Cell>(){
             public int compare(Cell a, Cell b) {
                 return a.height - b.height;
             }
         });

         int m = heightMap.length;
         int n = heightMap[0].length;
         boolean[][] visited = new boolean[m][n];

         for (int i = 0; i< m ;i++){
             visited[i][0] = true;
             visited[i][n-1] = true;
             queue.offer(new Cell(i, 0, heightMap[i][0]));
             queue.offer(new Cell(i, n-1, heightMap[i][n-1]));
         }

         for (int i = 0; i< n ;i++){
             visited[0][i] = true;
             visited[m-1][i] = true;
             queue.offer(new Cell(0, i, heightMap[0][i]));
             queue.offer(new Cell(m-1, i, heightMap[m-1][i]));
         }

         int[][] dirs = new int[][]{{-1,0},{1,0},{0,-1},{0,1}};
         int res = 0;
         while(!queue.isEmpty()){
             Cell cell = queue.poll();
             for(int[] dir : dirs){
                 int row = cell.row + dir[0];
                 int col = cell.col + dir[1];
                 if(row >= 0 && row < m && col >=0 && col < n && !visited[row][col]){
                     visited[row][col] = true;
                     res += Math.max(0,cell.height - heightMap[row][col]);
                     queue.offer(new Cell(row,col,Math.max(heightMap[row][col],cell.height)));
                 }
             }
         }
         return res;
     }

     public static void main(String[] args){
         int[][] heightMap={{1,4,3,1,3,2},{3,2,1,3,2,4},{2,3,3,2,3,1}};
         trapRainWater(heightMap);
     }
 }

																																																																																																																																																																																																	int[][] heightMap={{1,4,3,1,3,2},{3,2,1,3,2,4},{2,3,3,2,3,1}};
																																																																																																																																																																																																																																															trapRainWater(heightMap);    }}