每年的 1024 程序员节日,黑马程序员都会举办大型的庆祝活动。今年的程序员节也不例外,每个班级的同学都发了橙子。
班级里有 nn 名同学从前到后排成一排,且已经得知了这些同学的成绩,其中第 ii 名同学的成绩是 a_ia
i
。班主任想根据同学们上个阶段的考试成绩来评定发橙子的数量。为了激励成绩优秀同学,发橙子时需要满足如下要求:
相邻同学中成绩好的同学的橙子必须更多。若相邻的同学成绩一样,则它们分到的数量必须平等。
每个同学至少分配一个橙子
由于预算有限,班主任希望在符合要求的情况下发出尽可能少的橙子。请问,至少需要准备多少橙子呢?
输入格式
第一行是一个不超过 10^610
6
正整数 nn,表示学生数量。
接下来一行有 nn 个不超过 10^910
9
的非负整数 a_ia
i
,表示第 ii 个同学的成绩。
输出格式
输出答案,也就是需要最少准备多少个橙子。
输入输出样例
输入 #1复制
5
3 4 5 4 3
输出 #1复制
9
说明/提示
样例解释:
每位同学拿到的橙子的数量分别是 1,2,3,2,11,2,3,2,1,所以至少需要准备 99 个。
解题报告:看别的大佬的贪心思路就是先从左往右遍历,先把所有小朋友的橙子数+1,从左往右遍历,如果右边大于左边,右边等于左边+1,那么遍历完右边就满足啦,然后从右往左遍历,如果左边大于右边,取max,等于的时候也要取max(为了两边的都满足)。
#include<iostream>
#include<algorithm>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<vector>
#include<queue>
#include<set>
#define IL inline
#define x first
#define y second
typedef long long ll;
using namespace std;
const int N=1e6+5;
int a[N];
int t[N];
long long ans;
int main()
{
int n;
cin>>n;
for(int i=1;i<=n;i++)
cin>>a[i],t[i]=1;
for(int i=2;i<=n;i++)
if(a[i]>a[i-1]) t[i]=t[i-1]+1;
else if(a[i]==a[i-1]) t[i]=t[i-1];
for(int i=n;i>=1;i--)
if(a[i-1]>a[i]) t[i-1]=max(t[i-1],t[i]+1);
else if(a[i-1]==a[i]) t[i-1]=max(t[i-1],t[i]);
long long ans=0;
for(int i=1;i<=n;i++)
ans+=t[i];
cout<<ans<<endl;
return 0;
}