题目描述
一组正整数,分别表示由正方体叠起的柱子的高度。若某高度值为x,表示由x个正立方的方块迭起(如下图,0<=x<=5000)。找出所有可能积水的地方(图中蓝色部分),统计它们可能积水的面积总和(计算的是图中的横截面积。一个立方体的位置,为一个单位面积)。
如图:柱子高度变化为 0 1 0 2 1 2 0 0 2 0
图中蓝色部分为积水面积,共有6个单位面积积水。
输入格式
两行,第一行n,表示有n个数(3<=n<=10000)。第2行连续n个数表示依次由正方体迭起的高度,保证首尾为0。
输出格式
一个数,可能积水的面积。
输入输出样例
输入 #1
10 0 1 0 2 1 2 0 0 2 0
输出 #1
6
代码
#include<cmath>
#include<cstdio>
#include<cstring>
#include<iostream>
#include<algorithm>
using namespace std;
const int N=10010;
int n,ans,a[N];
int main() {
scanf("%d",&n);
for(int i=1; i<=n; i++)
scanf("%d",&a[i]);
int l=1,r=n;
while(a[l]==0)
l++;
while(a[r]==0)
r--;
int tmp=0,j=l,k=l+1;
while(k<=r+1) {
if(a[k]<a[j]) {
tmp+=a[k];
k++;
} else {
ans+=(k-j-1)*a[j]-tmp;
tmp=0;
j=k;
k++;
}
}
tmp=0,j=r,k=r-1;
while(k>=l-1) {
if(a[k]<=a[j]) {
tmp+=a[k];
k--;
} else {
ans+=(j-k-1)*a[j]-tmp;
tmp=0;
j=k;
k--;
}
}
printf("%d\n",ans);
return 0;
}