407. Trapping Rain Water II

题目：Given an m x n matrix of positive integers representing the height of each unit cell in a 2D elevation map, compute the volume of water it is able to trap after raining.

给定一个m x n的矩阵，表示一个2维的地图，矩阵里的数字都是正整数，表示每个单位的高度。要求计算这个地图里能储存的水量是多少。

题意：首先要确定什么情况下才会储存水量，只要边缘的长度大于自身的长度就会储存水量。储存水量的多少又是怎么决定的呢？根据伟大的木桶效应的原理，储存水量的多少决定在于最短的那块木板，也就是边缘最矮的那块。把题目要求的东西理解清楚后才可以开始做题。

思路：这道题搜索边缘当然就要从最外面开始找起，前面说了要找最短的木板，所以我们把最外面一圈找出来之后要进行排序，找到最短的边缘。基于这个特性，使用优先队列存储最为合适，把边缘的cell放入队列就会自动排好序，队列的头就是最短木板。另外也要用一个二维数组来存储表示哪些cell已经被搜索过了，下一层的搜索就不搜索这些cell。每一层的边缘首先取出最短的那条边，以此为基点缩小边缘的范围，直到所有的cell都被搜索过，在每次搜索当前最短边的时候，要先更新当前最短边的长度，将这条边取出来，新搜索到的边放进去，如果新搜索的边比当前最短边长度还要短，那必然会比边缘的任意一条边都短，那这个cell就可以储存水量，储存水量的多少等于当前边缘最短边的长度减去这个cell的高度。

class Solution {
public:
    int trapRainWater(vector<vector<int>>& heightMap)
    {
        int m = heightMap.size();//特殊情况判断
        if(m == 0)
            return 0;
        int n = heightMap[0].size();
        if(n == 0)
            return 0;
        priority_queue<cell> q;//优先队列存储边缘
        bool visit[120][120] = {0};//记录访问过的cell的二维数组
        for(int i = 0; i < m; ++i)
        {
            for(int j = 0; j < n; ++j)
            {
                if(i == 0 || j == 0 || i == m-1 || j == n-1)//检测边缘放入队列
                {
                    cell temp(i, j, heightMap[i][j]);
                    q.push(temp);
                    visit[i][j] = true;
                }
            }
        }
        int ans = 0;
        int dx[4] = {-1, 0, 1, 0};//边缘的四个方向
        int dy[4] = {0, -1, 0, 1};
        int maxHeight = -1;
        while(!q.empty())
        {
            cell c = q.top();//取出当前边缘中最短的边
            q.pop();
            int h = c.height;
            maxHeight = max(maxHeight, h);//更新新边缘的最短长度
            for(int i = 0; i < 4; ++i)
            {
                int new_x = c.x + dx[i];
                int new_y = c.y + dy[i];
                if(new_x >= 0 && new_y > 0 && new_x < m && new_y < n && visit[new_x][new_y] == false)
                {
                    visit[new_x][new_y] = true;
                    q.push(cell(new_x, new_y, heightMap[new_x][new_y]));//将新边缘放入队列
                    if(maxHeight > heightMap[new_x][new_y])
                        ans += maxHeight - heightMap[new_x][new_y];//增加储存的水量
                }
            }
        }
        return ans;
    }
private:
    struct cell
    {
        int x;
        int y;
        int height;
        cell(int a, int b, int h)
        {
            x = a;
            y = b;
            height = h;
        }
        friend bool operator< (cell n1, cell n2)
        {
            return n1.height > n2.height;
        }
    };
};