[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[] = {-, , , };
int dy[] = {, , , -};
class Solution {
public: int trapRainWater(vector<vector<int>>& h) {
int n = h.size();
if(n == ) return ;
int m = h[].size();
vector<vector<bool> > vis(n, vector<bool>(m, ));
priority_queue<pair<int, pair<int, int> > > q;
for (int i = ; i < n; i++) {
for (int j = ; j < m; j++) {
if(i == || j == || i == n - || j == m - ) {
vis[i][j] = ;
q.push({-h[i][j], {i, j}});
}
}
}
long long res = ;
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 = ; i < ; i++) {
int x = ux + dx[i];
int y = uy + dy[i];
if(x < || y < || 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] = ;
q.push({-h[x][y],{x, y} });
}
}
return res;
}
};
   int a[][];
bool v[][];
int dx[] = {, -, , };
int dy[] = {, , , -}; class Solution {
public:
bool in(int x, int y, int r, int c) {
return <= x && x < r && <= y && y < c;
}
int trapRainWater(vector<vector<int>>& h) {
priority_queue<pair<int, pair<int, int> > > q;
int m = h.size();
if(m == ) return ;
int n = h[].size(); memset(a, , sizeof a);
memset(v, , sizeof v);
for (int i = ; i < m; i++) {
for (int j = ; j < n; j++) {
if(i == || j == || i == m - || j == n - ) {
q.push(make_pair(-h[i][j], make_pair(i, j)));
a[i][j] = h[i][j];
v[i][j] = ;
}
}
}
// 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 = ; k < ; 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] = ;
q.push(make_pair(-a[nx][ny], make_pair(nx, ny)));
}
}
}
int ans = ;
for (int i = ; i < m; i++) {
for (int j = ; j < n; j++) {
ans += a[i][j] - h[i][j];
// printf("%d ", a[i][j]);
}
// printf("\n");
}
return ans;
}
};
上一篇:[LeetCode] 42. Trapping Rain Water 收集雨水


下一篇:Uncaught ReferenceError: console is not defined