程序员面试金典 面试题 17.21. 直方图的水量
题目链接
思路分析
通过从左向右遍历,从右向左遍历求得相对位置的最大值,然后从左向右遍历用他俩的最小值减去高度累加即可。
class Solution {
public:
int trap(vector<int>& height) {
std::ios::sync_with_stdio(false);
if(height.empty()) return 0;
int n = height.size();
vector<int> left(n), right(n);
left[0] = height[0], right[n - 1] = height[n - 1];
for(int i = 1; i < n; i++)
left[i] = max(left[i - 1], height[i]);
for(int i = n - 2; i >= 0; i--)
right[i] = max(right[i + 1], height[i]);
int res = 0;
for(int i = 0; i < n; i++)
res += min(left[i], right[i]) - height[i];
return res;
}
};