【提高】房屋积水c++

描述

乌龟家的屋顶是凹凸不平的，所以每次雨后都会积水。为了知道屋顶是否会在暴雨后塌掉，他把屋顶的形 状给了你，希望你帮他计算暴雨后屋顶的积水总量。

乌龟的屋顶由顺次排在同一水平线上的 n 个宽度为 1、高度为整数 (分别给出) 的瓦片组成。例如给定n = 5，瓦片的高度分别为 4, 2, 3, 5, 1，屋顶可以画在下图所示的网格中，灰色格子为瓦片。

暴雨过后，如果一个方格向左右两侧延伸都能到达瓦片占据的方格，它就会积水。所以图中波浪线格子在暴雨后会积水，屋顶的积水方格总数为 3。

输入描述

两个整数 n, R1，表示屋顶的宽度和生成数列的首项。从左向右数第 i ( 1<=i<=n )个瓦片的高度 ai = Ri mod 10

试题中使用的生成数列 R 定义如下：整数 0 ≤ R1 < 201701 在输入中给出。对于 i > 1，Ri = (Ri−1 × 6807 + 2831) mod 201701。

输出描述

一个整数，表示暴雨后屋顶积水方格的总数

用例输入 1 

10 1

用例输出 1 

23

提示

数据规模 1 ≤ n ≤ 100

来源
2017江苏省青少年信息学奥林匹克竞赛复赛

来源

省赛 动态规划

代码

#include<bits/stdc++.h>
#define maxn 10005
using namespace std;
int r[maxn],a[maxn],n,ans=0;
int main()
{
      cin>>n>>r[1];
    a[1]=r[1]%10;
    for(int i=2;i<=n;i++)
    {
        r[i]=(r[i-1]*6807+2831)%201701;
        a[i]=r[i]%10;//生成数列 
    }
    for(int i=1;i<=n;i++)
    {//求左右端点
        int z=INT_MIN,y=INT_MIN;//设初值
        for(int j=i;j>0;j--)
            if(a[j]>z)    z=a[j];
        for(int j=i;j<=n;j++)
            if(a[j]>y)    y=a[j];
        ans+=min(z,y)-a[i]; 
    }
    cout<<ans;
    return 0;
}