7-56 小H分糖果 (20 分)

#include<bits/stdc++.h>
#define ll long long
#define INF 0x7f7f7f7f //2139062143
#define llINF 9223372036854775807
#define IOS ios::sync_with_stdio(0); cin.tie(0); cout.tie(0);
using namespace std;

int n;
int num[100007],cas[100007];

int main()
{
    IOS
    cin>>n;
    for(int i=1;i<=n;i++)
    {
        cin>>num[i];
        cas[i]=0;
    }
    for(int i=1;i<=n;i++)
    {
        bool flag=1;
        if(i>1&&num[i-1]<num[i]) flag=0;
        if(i<n&&num[i+1]<num[i]) flag=0;
        if(flag) cas[i]=1;
    }
    ll ans=0;
    for(int i=2;i<=n;i++)
    {
        if(num[i]>num[i-1]) cas[i]=max(cas[i],cas[i-1]+1);
    }
    for(int i=n-1;i>0;i--)
    {
        if(num[i]>num[i+1]) cas[i]=max(cas[i],cas[i+1]+1);
    }
    for(int i=1;i<=n;i++) ans+=cas[i];
    cout<<ans<<endl;
}