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<algorithm>
#include<iostream>
#include<cstring>
#include<cstdio>
#include<cmath>
#include<queue>
using namespace std;
int n,a[10610],l,r,ans=0,tmp=0,j,k;
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
		scanf("%d",&a[i]);
	}
    l=1;
	while(a[l]==0){
		l++;
	}
    r=n;
	while(a[r]==0){
		r--;
	}
    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",ans);
    return 0;
}