老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
示例 1:
输入: [1,0,2]
输出: 5
解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
示例 2:
输入: [1,2,2]
输出: 4
解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
分类讨论
比前者大【暂定此位置糖果为前者+1,这个糖果可能因为后面孩子太少而增多,不过前面的不受影响n=1】
比前者相等【暂定此位置糖果为1,这个糖果可能因为后面孩子太少而增多,不过前面的不受影响n=1】
比前者小
若前者等于2,暂定此位置糖果为1,且影响位置的个数也+1【4 3 2->4 3 2 1】
若前者大于2,则暂定此位置糖果为1,且记录上一个孩子糖果个数【3->(3) 1,当1受影响变成2时,需要将3连起来(3) 1->(3) 2 1【3 2 1】】
若前者等于1,说明此位置孩子糖果太少了,要将前面受影响位置都+1,并暂定此位置糖果为1,且影响位置的个数也+1【3 2 1->4 3 2 1】
c++代码:
class Solution
{
puclic:
int candy(vector&ratings)
{
if(ratings.empty())return 0;
int ans=1,t=1,le=0,n=1;
for(int i=1;i<ratings.size();++i)
if(ratings[i]>ratings[i-1])
//n为受影响个数,ans为所需糖果数量,t为上一个孩子所需糖果果个数,le则用于判断其后一个孩子是否需要由1变成2。
n=1,ans+=++t,le=0;
else if(t2)
++n,ans+=–t;
else if(t>2)
le=t-2,t=n=1,++ans;
else
{
ans+=++n;
if(le&&!–le)++n;//即le1时
}
return ans;
}
};