题目描述
一组正整数,分别表示由正方体迭起的柱子的高度。若某高度值为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<cstring>
#include<iostream>
#include<algorithm>
#include<cmath>
#include<vector>
using namespace std;
const int N=10000+5;
int ans=0,num[N],n,maxn,pos=1,now;
int main()
{
scanf("%d",&n);
for (int i=1;i<=n;i++)
scanf("%d",&num[i]);
maxn=num[1];now=num[1];
for (int i=2;i<=n;i++)
if (num[i]>maxn) {maxn=num[i];pos=i;}
for (int i=2;i<pos;i++)
{
now=max(now,num[i]);
ans+=now-num[i];
}
now=num[n];
for (int i=n-1;i>=pos;i--)
{
now=max(now,num[i]);
ans+=now-num[i];
}
printf("%d\n",ans);
return 0;
}