接雨水

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。

示例:

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
思路:先找最高的,一分为二,让后找递增的序列,cur表示前一个最高的柱子,i表示当前柱子,只有当前柱子比cur要矮的时候,才会蓄水。如果后面一个柱子比当前高,则cur=i;直到最高的柱子。

#include<iostream>
#include<vector>
#include<algorithm>
#include<map>
#include<string>
#include<iomanip>
using namespace std;
class Solution {
public:
int trap(vector<int>& height) {
if (height.size()<3) return 0;
int max = 0;
int res = 0, cur = 0;
for (int i = 1; i<height.size(); ++i) {
if (height[i]>height[max])
max = i;
}
for (int i = 0, cur = 0; i<max; ++i) {
if (height[i]>height[cur]) {
cur = i;
}
else {
res += height[cur] - height[i];
}
}
for (int i = height.size() - 1, cur = height.size()-1; i>max; --i) {
if (height[i]>height[cur]) {
cur = i;
}
else {
res += height[cur] - height[i];
}
}
return res;
}

};

int main()
{
Solution solution;
vector<int> nums={ 4,2,3 };
int K=solution.trap(nums);
return 0;
}

posted @ 2019-03-18 20:30  zzas12345  阅读(330)  评论(0编辑  收藏  举报