题目描述
小H在吃过一次烤竹鼠之后,觉得非常美味,也动了自己养殖竹鼠的心思,正好家附近有很多竹子,于是在家旁建了个养殖场,他把养殖场隔出来N个隔间并且编好了号,每个隔间内养一只竹鼠。但竹鼠的特性非常奇怪,在喂食的时候,体重较大的竹鼠如果看到,左右相邻的隔间有竹鼠比它体重更轻却喂了更多的食物,大竹鼠就会心情郁闷而影响到它的健康。
为了科学高效的养殖,小H决定喂食时遵循两个规矩:
1,每个竹鼠最少要喂1片竹子
2,相邻的竹鼠中,如果体重不同,则体重更高的竹鼠必须喂更多的竹子请问小H最少需要为竹鼠们准备多少片竹子呢?
输入
第一行一个整数N,表示竹鼠数,其中0<N≤50000;
第二行N个数表示不同竹鼠的体重,以空格隔开,每个数不超过50000。
输出
输出一个数,表示最少需要准备的竹片数
样例输入 Copy
3
1 2 2
样例输出 Copy
4
提示
一共有3只竹鼠,它们的体重分别是1 2 2,那么最少需要喂的竹片数量为1+2+1=4片
对于5%的数据,0<N≤10;
对于30%的数据,0<N≤2000;
对于100%的数据,0<N≤50000。
#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
typedef long long ll;
int a[N];
ll res[N];
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++) cin>>a[i];
for(int i=1;i<=n;i++) res[i]=1;
ll sum=0;
for(int i=2;i<=n;i++)
{
if(a[i]>a[i-1]) res[i]=res[i-1]+1;
}
for(int i=n-1;i>=1;i--)
{
if(a[i]>a[i+1]&&res[i]<=res[i+1]) res[i]=res[i+1]+1;
}
for(int i=1;i<=n;i++ ) sum+=res[i];
cout<<sum<<endl;
}