[leetcode] 407. Trapping Rain Water II

https://leetcode.com/contest/6/problems/trapping-rain-water-ii/

看到这题，我很高兴，因为我做过！哈哈！其实我现在也写不出来，知道大概思想。

这题是google apactest 2017 round A 的第二题。https://code.google.com/codejam/contest/11274486/dashboard#s=p1

然后，简单一次调试，就ac了。不知道这算不算作弊，做完，我就看见什么instruction，说是可以看别人代码，然后举报作弊，有点怕！

分析：这题算是动态规划吧，读懂题意后，（其实这个题目描述的不清楚啊，图的示例倒是挺好），可以观察，可以从外圈往里圈缩，因为这个过程墙的高度是非递减的，然后，你应该想到用优先队列（priority_queue或者set，又是讨厌的set），先把外圈所有点加入，然后取出高度最小的点，更新四周的点，注意标记这个点是否访问过，这个过程中记录墙增加的高度就是最后的积水量。

哎！咸鱼也是有梦想的！

int dx[] = {-1, 0, 1, 0};
int dy[] = {0, 1, 0, -1};
class Solution {
public:

int trapRainWater(vector<vector<int>>& h) {
    int n = h.size();
    if(n == 0) return 0;
    int m = h[0].size();
    vector<vector<bool> > vis(n, vector<bool>(m, 0));
    priority_queue<pair<int, pair<int, int> > > q;
    for (int i = 0; i < n; i++) {
        for (int j = 0; j < m; j++) {
            if(i == 0 || j == 0 || i == n - 1 || j == m - 1) {
                vis[i][j] = 1;
                q.push({-h[i][j], {i, j}});
            }
        }
    }
    long long res = 0;
    while(!q.empty()) {
        int u = -q.top().first;
        int ux = q.top().second.first;
        int uy = q.top().second.second;
        q.pop();
        //cout << ux << " " << uy << " " << u << endl;
        for (int i = 0; i < 4; i++) {
            int x = ux + dx[i];
            int y = uy + dy[i];
            if(x < 0 || y < 0 || x >= n || y >= m || vis[x][y])
                continue;
            if(h[x][y] < u) {
                res += u - h[x][y];
                h[x][y] = u;
            }
            vis[x][y] = 1;
            q.push({-h[x][y],{x, y} });
        }
    }
    return res;
}
};

  int a[120][120];
    bool v[120][120];
    int dx[] = {1, -1, 0, 0};
    int dy[] = {0, 0, 1, -1};
   
class Solution {
public:
 bool in(int x, int y, int r, int c) {
        return 0 <= x && x < r && 0 <= y && y < c;
    }
   int trapRainWater(vector<vector<int>>& h) {
        priority_queue<pair<int, pair<int, int> > >    q;
        int m = h.size();
        if(m == 0) return 0;
        int n = h[0].size();
       
        memset(a, 0, sizeof a);
        memset(v, 0, sizeof v);
        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) {
                    q.push(make_pair(-h[i][j], make_pair(i, j)));
                    a[i][j] = h[i][j];
                    v[i][j] = 1;
                }
            }
        }
        // cout << n << " " << m << endl;
        while(q.size()) {
            pair<int, pair<int, int> > u = q.top();
            q.pop();
            int x = u.second.first;
            int y = u.second.second;
            for (int k = 0; k < 4; k++) {
                int nx = x + dx[k];
                int ny = y + dy[k];
                if (in(nx, ny, m, n) && !v[nx][ny]) {
                    if (h[nx][ny] < a[x][y]) {
                        a[nx][ny] = a[x][y];
                    } else {
                        a[nx][ny] = h[nx][ny];
                    }
                    v[nx][ny] = 1;
                    q.push(make_pair(-a[nx][ny], make_pair(nx, ny)));
                }
            }
        }
        int ans = 0;
        for (int i = 0; i < m; i++) {
            for (int j = 0; j < n; j++) {
                ans += a[i][j] - h[i][j];
//                printf("%d ", a[i][j]);
            }
//            printf("
");
         }
         return ans;
    }
};