题目描述:
有N个小朋友站在一排,每个小朋友都有一个评分
你现在要按以下的规则给孩子们分糖果:
每个小朋友至少要分得一颗糖果
分数高的小朋友要他比旁边得分低的小朋友分得的糖果多
你最少要分发多少颗糖果?
思路
- 初始化每人一颗糖果
- 正序遍历 若后面的人评分高于前面 糖数比前面人多一颗
- 倒序遍历 若前面的人评分高于后面且糖不大于后面 糖数比后面人多一颗
#
#
# @param ratings int整型一维数组
# @return int整型
#
class Solution:
def candy(self , ratings ):
# write code here
result = [1]*len(ratings)
for i in range(1,len(ratings)):
if ratings[i]>ratings[i-1] :
result[i] = result[i-1]+1
for i in range(len(ratings)-2,-1,-1):
if ratings[i]>ratings[i+1] and result[i] <= result[i+1]:
result[i] = result[i+1]+1
return sum(result)
r = [1,2,3,4,9,2,1,4,2,66,2]
A = Solution()
print(A.candy(r))