题目描述
n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。
你需要按照以下要求,给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。 相邻两个孩子评分更高的孩子会获得更多的糖果。 请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
输入格式
第一行是n的个数
第二行 是整数数组 ratings 表示每个孩子的评分
输出格式
需要准备的 最少糖果数目
输入输出样例
输入 #1复制
3 1 0 2
输出 #1复制
5
输入 #2复制
3 1 2 2
输出 #2复制
4
说明/提示
n=ratings.length
1≤n≤2∗10^4
0≤ratings[i]≤2∗10^4
输入:1 2 2
输出:4
解释:你可以分别给第一个、第二个、第三个孩子分发 1、2、1 颗糖果。 第三个孩子只得到 1 颗糖果,这满足题面中的两个条件。
代码如下:
#include<stdio.h>
int main()
{
int n,rating[20010],l[20010],r[20010];
scanf("%d",&n);
for(int i=0;i<n;i++)
scanf("%d",&rating[i]);
for(int i=0;i<n;i++)//仅看左侧找糖果数量
{
if(rating[i]>rating[i-1]&&i>0)
l[i]=l[i-1]+1;
else l[i]=1;
}
for(int j=n-1;j>=0;j--)//仅看右侧找糖果数量
{
if(j<n-1&&rating[j]>rating[j+1])
r[j]=r[j+1]+1;
else
r[j]=1;
}
int sum=0;
for(int i=0;i<n;i++)//为同时满足左右要求取最大值
{
if(l[i]>=r[i])sum=sum+l[i];
else
sum=sum+r[i];
}
printf("%d",sum);
return 0;
}