Luogu 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 <bits/stdc++.h>
#define N 100005
#define ll long long
#define ull unsigned long long
#define il inline
#define rg register
using namespace std;
int a[10001];
int main()
{
    cin.sync_with_stdio(false);
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++)
        cin >> a[i];
    int begin = 0, end = 0, begini, endi;
    int ans = 0;
    for (int i = 1; i <= n; i++)
    {
        if (a[i] > begin)
        {
            begin = a[i], begini = i;
            int j = i + 1;
            while (a[j] < begin && j <= n)
                j++;
            if (j >= n)
            {
                int maxx = 0, maxi = 0;
                for (j = i + 1; j <= n; j++)
                {
                    if (a[j] > maxx)
                    {
                        maxx = a[j];
                        maxi = j;
                    }
                    if (a[j] == maxx)
                        maxi = max(maxi, j);
                }
                j = maxi;
            }
            int h = min(a[i], a[j]);
            for (int k = i + 1; k < j; k++)
            {
                if (a[k] <= h)
                    ans += h - a[k];
            }
            i = j - 1;
            begin = 0;
        }
    }
    cout << ans << endl;

    system("pause");
    return 0;
}