【LeetCode】贪心算法--分发糖果（135）

今日题目

老师想给孩子们分发糖果，有N个孩子站成了一条直线，老师会根据每一个孩子的表现，预先给他们评分。
你须要按照如下要求，帮助老师给这些孩子分发糖果：
每一个孩子至少分配到 1 个糖果。相邻的孩子中，评分高的孩子必须得到更多的糖果。那么这样下来，老师至少须要准备多少颗糖果呢？

示例 1:
输入: [1,0,2]
输出: 5
解释: 你能够分别给这三个孩子分发 二、一、2 颗糖果。
示例 2:
输入: [1,2,2]
输出: 4
解释: 你能够分别给这三个孩子分发 一、二、1 颗糖果。第三个孩子只获得 1 颗糖果，这已知足上述两个条件。

题目分析

对于这个问题，主要的条件是要相邻孩子，得分高的拿的糖果要多（不包括相等得分），还有就是每一个人最少都要有一个。
解决这个问题咱们仍是采用贪心算法，首先初始化每一个人分配的糖果数量都是1，而后这个算法须要遍历两遍，第一遍从左往右遍历，若是当前孩子的分数大于前一个孩子的分数，则当前孩子获得的糖果在前一个孩子的基础上加1；而后，第二遍从右往左遍历，若是当前孩子的分数大于他右边孩子的分数，而且他的糖果不比他右边孩子多，则糖果数在他基础上加1；最后，将全部孩子的糖果数相加便可。
第二次遍历是为了处理数组中降序和有出现相邻孩子分数相同的状况。

代码实现

class Solution:

def candy(self, ratings):
   """
   :type ratings: List[int]
   :rtype: int
    """
   num = [1]*len(ratings)
   for i in range(0,len(ratings)-1):
   if ratings[i+1] > ratings[i]:
       num[i+1] = num[i] + 1     
   for i in range(len(ratings)-1,0,-1):
       if ratings[i] < ratings[i-1]:
           num[i-1] = max(num[i]+1,num[i-1])

   return sum(num)

写在最后

寒假已经跟一个小伙伴商量好一块儿刷题，感兴趣的小伙伴也能够加入咱们，但愿你们天天能够交流刷题的心得，为了保证质量我会控制人数，大约在12人左右。

推荐阅读：
Python骚操做 | 用Python来P图