描述
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
输入
一个整数n,随后是n个非负整数,分别表示柱子的高度
1 ≤ n ≤ 2 * 104
每根柱子高度≤ 105
输出
能接到的雨水量
样例输入
12 0 1 0 2 1 0 1 3 2 1 2 1
样例输出
6
提示
样例解释:该情形见下图,黑色表示柱子,蓝色表示能接到的雨水
解题分析
首先,由木桶原理,我们知道一个由很多块木板围在一起的木桶所能装下的水的量将会由最短的那块木板决定。我们不妨先从左往右看,我们把能够容纳雨水的一块块区域看作一个个木桶,那么这个木桶其实只有两块木板,左边那块和右边这块,而且我们不妨先假定左边这块木板为最短木板的情况,那么我们从左往右遍历柱子,如果遇到比我们假定的最短木板高的木板的情况,说明可以储水了,那么我们开始储水,然后更新我们的最短木板的长度;同理,另一种情况,那就是右边这块木板短的情况,我们从右往左遍历,假定右边的木板短,那么我们遇到更高的木板的时候,就开始储水,然后同理更新我们的更短木板的长度。最后,我们输出我们储水的总量即可。事实上,我们也可以用双指针去简化我们的算法流程,那样就可以只遍历一次。但是实际上都是n的级别,而两次遍历会更加的简洁。
代码实现
#include <iostream>
#include <cstdio>
#include <queue>
#include <bitset>
#include <map>
#include <set>
#include <algorithm>
#include <cstdlib>
#include <unordered_map>
#include <unordered_set>
#include <cmath>
#include <vector>
#include <sstream>
using namespace std;
int n;
int rain[20005];
int main(){
scanf("%d",&n);
for(int i=0;i<n;i++){
scanf("%d",&rain[i]);
}
int l=0,r=n-1;
int count=0;
int pos=0;
for(int i=1;i<n;i++){
if(rain[i]>rain[pos]){
for(int j=pos+1;j<i;j++){
count+=rain[pos]-rain[j];
}
pos=i;
}
}
pos=n-1;
for(int i=n-1;i>=0;i--){
if(rain[i]>rain[pos]){
for(int j=pos-1;j>i;j--){
count+=rain[pos]-rain[j];
}
pos=i;
}
}
printf("%d\n",count);
return 0;
}