洛谷 P1318 积水面积

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<cstdio>
#include<cstdlib>
#include<iostream>
#include<algorithm>
using namespace std;
int n,maxn,ans;
int a[5010],l[5010],r[5010];
int main(){
    scanf("%d",&n);
    for(int i=1;i<=n;i++){
        scanf("%d",&a[i]);
        maxn=max(maxn,a[i]);    
    }
    for(int i=1;i<=n;i++)
        l[i]=max(a[i],l[i-1]);
    for(int i=n;i>=1;i--)
        r[i]=max(a[i],r[i+1]);
    for(int i=1;i<=maxn;i++){
        for(int j=1;j<=n;j++)
            if((l[j]>=i)&&(r[j]>=i)&&(a[j]<i))    ans++;
    }
    cout<<ans;
}
/*
12
0 3 2 0 0 2 0 3 1 1 2 0
*/