# LeetCode 42 Trapping Rain Water

512人阅读 评论(0)

Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.

For example,
Given [0,1,0,2,1,0,1,3,2,1,2,1], return 6.

The above elevation map is represented by array [0,1,0,2,1,0,1,3,2,1,2,1]. In this case, 6 units of rain water (blue section) are being trapped. Thanks Marcos for contributing this image!

public class Solution {
public int trap(int[] height) {
int maxHeight = 0, len = height.length, pos = 0;
for(int i = 0; i < len; i ++) {
if(maxHeight < height[i]) {
pos = i;
maxHeight = height[i];
}
}
int ans = 0;
int curMax = 0, curArea = 0;
for(int i = 0; i < pos; i ++) {
if(height[i] > curMax) {
curMax = height[i];
}
else {
curArea += curMax - height[i];
}
}
ans += curArea;
curMax = 0;
curArea = 0;
for(int i = len - 1; i > pos; i --) {
if(height[i] > curMax) {
curMax = height[i];
}
else {
curArea += curMax - height[i];
}
}
ans += curArea;
return ans;
}
}

个人资料
等级：
访问量： 75万+
积分： 1万+
排名： 867
博客专栏
 算法题 文章：50篇 阅读：35147 Java学习笔记 文章：10篇 阅读：11888 LeetCode 文章：138篇 阅读：80036