P1318 积水面积

题目描述

一组正整数，分别表示由正方体叠起的柱子的高度。若某高度值为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;
}