1550 - 【提高】房屋积水

题目描述

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

乌龟的屋顶由顺次排在同一水平线上的 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。

输出

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

样例

输入

复制

10 1

输出

复制

23

说明

数据规模 1 ≤ n ≤ 100

直接上代码：

#include<bits/stdc++.h>
using namespace std;
int a[110],l[110];
int main(){int n,r;cin>>n>>r; l[1]=r;a[1]=l[1]%10;for(int i=2;i<=n;i++){l[i]=(l[i-1]*6807+2831)%201701;a[i]=l[i]%10;}int ans=0;for(int i=0;i<=10;i++){for(int j=1;j<=n;j++){if(a[j]>i){int k=j+1;while(a[k]<=i&&k<=n)k++;if(k<=n)ans+=k-j-1;j=k-1;}}}cout<<ans<<endl;return 0;
}