n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。你需要按照以下要求,给这些孩子分发糖果:①每个孩子至少分配到 1 个糖果。②相邻两个孩子评分更高的孩子会获得更多的糖果。 请你给每个孩子分发糖果,计算并返回需要准备的 最少糖果数目 。
输入:ratings = [1,0,2] 输出:5 解释:你可以分别给第一个、第二个、第三个孩子分发 2、1、2 颗糖果。
思路:这道题目一定是要确定一边之后,再确定另一边,例如比较每一个孩子的左边,然后再比较右边,如果两边一起考虑一定会顾此失彼。
-
先确定右边评分大于左边的情况(从前向后遍历)
此时局部最优:只要右边评分比左边大,右边的孩子就多一个糖果,全局最优:相邻的孩子中,评分高的右孩子获得比左边孩子更多的糖果
局部最优可以推出全局最优。
-
再确定左孩子大于右孩子的情况(从后向前遍历)
遍历顺序这里有同学可能会有疑问,为什么不能从前向后遍历呢?
因为 rating[5]与rating[4]的比较 要利用上 rating[5]与rating[6]的比较结果,所以 要从后向前遍历。
如果从前向后遍历,rating[5]与rating[4]的比较 就不能用上 rating[5]与rating[6]的比较结果了。
-
那么本题采用了两次贪心的策略:
-
一次是从左到右遍历,只比较右边孩子评分比左边大的情况。
-
一次是从右到左遍历,只比较左边孩子评分比右边大的情况。
这样从局部最优推出了全局最优,即:相邻的孩子中,评分高的孩子获得更多的糖果。
-
class Solution:
def candy(self, ratings: List[int]) -> int:
ans = [1]*len(ratings)
for i in range(len(ratings)-2,-1,-1):
if ratings[i]>ratings[i+1]:
ans[i]=ans[i+1]+1
cnt = ans[0]
for i in range(1,len(ratings)):
if ratings[i]>ratings[i-1]:
# 第二趟遍历的时候,要保留两次遍历的最大值
ans[i]=max(ans[i-1]+1,ans[i])
cnt += ans[i]
return cnt