题目
来源:力扣(LeetCode)
链接:接雨水
分析
计算每个位置可以积累的水柱高度,需要分别计算位置i左右两边的所有柱子中的最高柱子的高度。
可以利用动态规划的思想,使用两个数组动态的存储从柱子0到柱子i的最大高度、从柱子n-1到柱子i的最大高度,分别记为max_left[],max_right[]。
如上图柱子1的高度为3,左边最高的柱子高度为5,右边最高的柱子高度为6,在柱子1中可以积水的高度为min(5,6)-3=2。
所有情况为下表
i | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 |
---|---|---|---|---|---|---|---|---|---|
max_left[i] | 5 | 5 | 5 | 5 | 5 | 6 | 6 | 6 | 6 |
max_right[i] | 6 | 6 | 6 | 6 | 6 | 6 | 5 | 2 | 2 |
min(max_left[i],max_right[i]) | 5 | 5 | 5 | 5 | 5 | 6 | 5 | 2 | 2 |
代码
int min(int a,int b){
return a<b?a:b;
}
int trap(int* height, int heightSize){
int* max_left=(int*)malloc(sizeof(int)*heightSize);
int* max_right=(int*)malloc(sizeof(int)*heightSize);
int sumWarter=0;
int max=0;
for(int i=0;i<heightSize;i++){
if(height[i]>max){
max=height[i];
}
max_left[i]=max;
}
max=0;
for(int j=heightSize-1;j>=0;j--){
if(height[j]>max){
max=height[j];
}
max_right[j]=max;
}
for(int i=0;i<heightSize;i++){
sumWarter+=min(max_left[i],max_right[i])-height[i];
}
free(max_left);
free(max_right);
return sumWarter;
}