蓝桥杯训练——[传智杯]程序员节发橙子

题目链接：https://www.dotcpp.com/oj/contest2415_problem1.html

#include <iostream>
#include <cmath>
#include <algorithm>
#include <cstring>
#include <vector>
#include <map>
#include <queue>
#include <cstdio>
#include <string>
#include <stack>
#include <set>
#define IOS ios::sync_with_stdio(false), cin.tie(0)
using namespace std;
typedef long long ll;
ll a[1000010];
//本题的思想：
//把每一个增序列接着再减序列看做一个整体，这样每个整体的两边必定分到一个橘子(最少)
//对每个整体：先从左边开始向中间分的橘子递增(至少不减)，再从右边向中间分的橘子递增
//这样分数处于中间分数的人会分到两种可能数量的橘子，取最多那一种可能
int main()
{
    IOS;
    ll n;
    cin>>n;
    ll min_id=1;
    for(ll i=1;i<=n;i++){
        cin>>a[i];
    }
    ll num=1,ans=1;
    for(ll i=2;i<=n;){//正向遍历
        if(a[i]>=a[i-1]){//每个整体的左半部分本身就是正向遍历
            if(a[i]==a[i-1])
                ans+=num;
            else ans+=(++num);
            i++;
        }
        else {//右半部分逆向遍历，需要统计右区间的左右端点下标
            //mid标记右区间的左界限，temp记录处于中间分数的人分到的最少的橘子
            ll mid=i-1,temp=num;
            while(i<=n&&a[i]<=a[i-1]){
               i++;
            } 
           num=1;//右区间分最少的人分到一个橘子
           ans++;
           ll j=i-2;//从右区间分数次最少的一个人还是向中间分数的人遍历(逆序)
           while(j>mid&&a[j]>=a[j+1]){
               if(a[j]==a[j+1])ans+=num;
               else ans+=(++num);
               j--;
           }
           //处于中间分数的人可能取到temp和num两种可能数量的橘子，取最大的
           //注意temp已经提前计入到ans中，需要进行作差操作
           if(temp<=num){
               while(a[j]==a[mid]){//处于中间分数的人可能有好几个
                   ans+=(num-temp+1);j--;
               }
           }
           num=1;//继续循环找新的“整体”
        }
    }
    cout<<ans<<endl;
    getchar();
    getchar();
    return 0;
}