题目大意
一直线上站了N个孩子,每个孩子都有一个属于自己的数字,现在按照如下规则给孩子分发糖果:每个孩子至少有一个糖果;相邻的孩子中数字比较大的那个拿的糖果也比较多。求最少要发掉多少个糖果。
解题思路
贪心法。
想象下,先从前面开始升序遍历,所有升序的就从1开始给,再从后面开始升序遍历,所有升序的就从1开始给,遇到所有需要改变值的节点,对比想要改成的值与之前的值,取最大的保证其对两边都是最大。
注意:
在数组一开始就为降序时,如下例,他的第1和第2个孩子会先分到同样的1个。要从后面开始扫描,将降序的较高孩子分得更多的糖果。
我的输入
[1,0,2]
我的标准输出
[1, 1, 2]
[2, 1, 2]
我的答案
5
代码
class Solution(object):
def candy(self, ratings):
""" :type ratings: List[int] :rtype: int """
n = len(ratings)
result = [1 for __ in range(n)]
for i in range(1, n):
if ratings[i-1] < ratings[i]:
result[i] = result[i-1] + 1
# 要从后面开始扫描,将降序的较高孩子分得更多的糖果。
for j in range(n-2, -1, -1):
if ratings[j] > ratings[j+1]:
result[j] = max(result[j], result[j+1] + 1)
return sum(result)