I Want Candy | ||||||
| ||||||
Description | ||||||
小八被拜托给参加校赛(个人赛)的同学们分发糖果,长老给了n个同学的最终得分,有以下两个要求: 每个同学都至少分得一块糖果; 对于相邻的两个同学,分数高的同学要比分数低的同学分得的糖果多。 本着不要浪费的原则,长老让小八计算出最少需要多少块糖果然后去买... “长老分明是在考验我小叮当”,还是帮帮小八吧...555... | ||||||
Input | ||||||
有多组输入数据,每组数据先输入一个整数n(1≤n≤105),表示有n个参加校赛的同学。接下来一行输入n个整数ai(0≤ai≤105),按顺序给出n位同学的得分。 | ||||||
Output | ||||||
对于每组输入数据,输出一个数字,表示最少需要的糖果数,每组输出占一行。 | ||||||
Sample Input | ||||||
3 1 2 2 | ||||||
Sample Output | ||||||
4 |
正向扫一遍反向扫一遍保留每一个位置的最大值。
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
#define N 100006
int num[N];
int a[N];
int main()
{
int n;
while(~scanf("%d",&n))
{
for(int i=0; i<n; i++)
{
scanf("%d",&a[i]);
}
memset(num,0,sizeof(num));
num[0]=1;
for(int i=1; i<n; i++)
{
if(a[i]>(a[i-1]))
{
num[i]=num[i-1]+1;
}
else
{
num[i]=1;
}
}
for(int i=n-2; i>=0; i--)
{
if(a[i]>a[i+1])
{
num[i]=max(num[i],num[i+1]+1);
}
}
int res=0;
for(int i=0; i<n; i++)
{
res+=num[i];
}
printf("%d\n",res);
}
}