这道题目是一道很有意思的题目,是老师给下朋友分糖的策略,题目如下:
这道题目的难点就在于,相邻的孩子中,评分高的孩子必须获得更多的糖果。我们必须把所有的小朋友左右都进行遍历一遍,然后根据评分来分配糖果,我们可以用如下策略:
首先我们给每个小朋友先发一颗糖,然后正向遍历一遍小朋友的评分,如果相邻的右边小朋友的评分比左边小朋友高,那么我们就给右边这个小朋友发一颗糖,然后将额外多发的糖的数量记录下来。同理我们反向遍历一遍这个评分列表,如果相邻的左边的小朋友的评分比右边高的话,我们就给左边的小朋友多发一颗糖,那么两个方向都遍历完成之后,我们将每一个小朋友的一颗糖,加上额外发的糖数,就是老师需要准备的糖数,问题就解决了。代码如下所示:
class Solution(object):
def candy(self, ratings):
"""
:type ratings: List[int]
:rtype: int
"""
length = len(ratings)
total = length
tmp = [0 for _ in range(length)]
for i in range(1, length):
if ratings[i] > ratings[i-1]:
tmp[i] = tmp[i-1] + 1
for i in range(length-2, -1, -1):
if ratings[i + 1] < ratings[i]:
tmp[i] = max(tmp[i], tmp[i+1] + 1)
total += sum(tmp)
return total
所以说我们需要双向遍历列表来分别判断相邻左右小朋友的关系,还需要判断当前额外的糖果数,和右边小朋友糖果数加一的结果取较大值,这就是这道题目的解法。