leetcode135：candy（python）

题目：

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。

你需要按照以下要求，给这些孩子分发糖果：

• 每个孩子至少分配到 1 个糖果。
• 相邻两个孩子（左右都要考虑到）评分更高的孩子会获得更多的糖果。
• 计算并返回需要准备的 最少糖果数目 。

解答：

先从左到右遍历一遍，然后反过来

 

class Solution(object):
    def candy(self,ratings):
        n = len(ratings)
        dp = [1 for i  in range(n)]
        sum = 0
        for i in range(n):
            if ratings[i]>ratings[i-1]:
                dp[i] = dp[i-1]+1


        for i in range(n-2,-1,-1):
            if ratings[i]>ratings[i+1]and dp[i]<dp[i+1]:
                dp[i]=dp[i+1]+1

        for i in range(n):
            sum += dp[i]
            return sum

#调用
s = Solution()
print(s.candy([1,0,2]))

运行结果与测试：

小结：

使用了贪心算法，贪心策略是：每次遍历只考虑一侧（局部）的大小关系