码迷,mamicode.com
首页 > 移动开发 > 详细

[leetcode] 407. Trapping Rain Water II

时间:2016-09-25 13:10:35      阅读:199      评论:0      收藏:0      [点我收藏+]

标签:

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),先把外圈所有点加入,然后取出高度最小的点,更新四周的点,注意标记这个点是否访问过,这个过程中记录墙增加的高度就是最后的积水量。

哎!咸鱼也是有梦想的!

 1   int a[120][120];
 2     bool v[120][120];
 3     int dx[] = {1, -1, 0, 0};
 4     int dy[] = {0, 0, 1, -1};
 5    
 6 class Solution {
 7 public:
 8  bool in(int x, int y, int r, int c) {
 9         return 0 <= x && x < r && 0 <= y && y < c;
10     }
11    int trapRainWater(vector<vector<int>>& h) {
12         priority_queue<pair<int, pair<int, int> > >    q;
13         int m = h.size();
14         if(m == 0) return 0;
15         int n = h[0].size();
16        
17         memset(a, 0, sizeof a);
18         memset(v, 0, sizeof v);
19         for (int i = 0; i < m; i++) {
20             for (int j = 0; j < n; j++) {
21                 if(i == 0 || j == 0 || i == m - 1 || j == n - 1) {
22                     q.push(make_pair(-h[i][j], make_pair(i, j)));
23                     a[i][j] = h[i][j];
24                     v[i][j] = 1;
25                 }
26             }
27         }
28         // cout << n << " " << m << endl;
29         while(q.size()) {
30             pair<int, pair<int, int> > u = q.top();
31             q.pop();
32             int x = u.second.first;
33             int y = u.second.second;
34             for (int k = 0; k < 4; k++) {
35                 int nx = x + dx[k];
36                 int ny = y + dy[k];
37                 if (in(nx, ny, m, n) && !v[nx][ny]) {
38                     if (h[nx][ny] < a[x][y]) {
39                         a[nx][ny] = a[x][y];
40                     } else {
41                         a[nx][ny] = h[nx][ny];
42                     }
43                     v[nx][ny] = 1;
44                     q.push(make_pair(-a[nx][ny], make_pair(nx, ny)));
45                 }
46             }
47         }
48         int ans = 0;
49         for (int i = 0; i < m; i++) {
50             for (int j = 0; j < n; j++) {
51                 ans += a[i][j] - h[i][j];
52 //                printf("%d ", a[i][j]);
53             }
54 //            printf("\n");
55          }
56          return ans;
57     }
58 };

 

[leetcode] 407. Trapping Rain Water II

标签:

原文地址:http://www.cnblogs.com/y119777/p/5905593.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!