贪心：leetcode135（困难）——分发糖果

最新推荐文章于 2024-08-24 19:27:15 发布

原创最新推荐文章于 2024-08-24 19:27:15 发布 · 151 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#unity #unity3d #贪心算法

题目分类专栏收录该内容

27 篇文章

订阅专栏

本文讲解如何使用Python实现糖果分配算法，通过两次迭代确保每个人至少一块糖果且满足相邻者的糖果数量差不超过一。关键步骤包括初次按从左到右分配更多，然后从右到左调整以平衡多出的糖果。

题目

在这里插入图片描述

题解

思路：
1.首先每个人至少一块糖果
2.循环两遍数组，从左往右，右边分高的至少比左边多一块；
从右往左，因为这个时候可能左边分比右边高的同时糖果比右边多，这个时候维持原状，糖果比右边少的话就要比右边多一个

class Solution:
    def candy(self, ratings: List[int]) -> int:
        n = len(ratings)
        nums = [1]*n    #每人最少有一块
        for i in range(0, n-1):     #右边分高的至少比左边多一块
            if ratings[i] < ratings[i+1]:
                nums[i+1] = nums[i]+1
        for i in range(n-1, 0, -1):     #左边分高的如果糖果没右边的多，就要分给比右边最少多一块
            if ratings[i-1] > ratings[i]:
                nums[i-1] = max(nums[i-1], nums[i]+1)
        return sum(nums)