LeetCode：135. Candy - Python

135.分发糖果

问题描述：

老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。

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

(1) 每个孩子至少分配到1个糖果。
(2) 评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。
那么这样下来，老师至少需要准备多少颗糖果呢？

示例:

输入：[1,0,2]
输出：5
解释：你可以分别给这三个孩子分发 2、1、2 颗糖果。

问题分析：

题目解法有多种，题目提示为 贪心和数组，先介绍双数组法。根据题目要求（2）可知评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。
(1) 我们可以先从左边向右边扫描，先满足每个孩子左边的条件
(2) 同理，在从右边向左边扫描，满足每个孩子右边的条件
(3) 最后根据取两者最大值即可满足要求

Python3实现：

# @Time   :2019/06/17
# @Author :Liu
# 双数组左右扫描法

class Solution:
    def candy(self, ratings):
        """
        :type ratings: List[int]
        :rtype: int
        """
        res = [1]*len(ratings)
        lbase, rbase = 1, 1

        for i in range(1, len(ratings)):  # 从左向右扫描
            lbase = lbase + 1 if ratings[i] > ratings[i-1] else 1
            res[i] = lbase

        for i in range(len(ratings)-2, -1, -1):  # 从右向左扫描
            rbase = rbase + 1 if ratings[i] > ratings[i+1] else 1
            res[i] = max(rbase, res[i])
        return sum(res)


if __name__ == '__main__':
    ratings = [1, 2, 2]
    print(Solution().candy(ratings))

一遍扫描方法

这个考虑峰值和谷点，稍微有点难理解，感兴趣的可以看下，下面是其他同学提交的优选代码可以参考：

# 他人提交的优秀代码段


class Solution:
    def candy(self, ratings):
        """
        :type ratings: List[int]
        :rtype: int
        """
        if len(ratings) <= 1: return len(ratings)
        candies = 1
        up = down = peak = 0
        for i in range(1, len(ratings)):  # 上升阶段
            if ratings[i - 1] < ratings[i]:
                up += 1
                peak = up
                candies += 1 + up
                down = 0
            elif ratings[i - 1] > ratings[i]:  # 下降阶段
                down += 1
                candies += 1 + down - (1 if peak >= down else 0)
                up = 0
            else:  # 如果相等
                peak = up = down = 0
                candies += 1
        return candies


if __name__ == '__main__':
    ratings = [1, 2, 2]
    ratings = [4, 5, 6, 5, 4, 3, 2, 1]
    print(Solution().candy(ratings))

声明： 总结学习，有问题或不当之处，可以批评指正哦，谢谢。

题目链接