Leetcode【1103】 Distribute Candies to People（Python版）

原创于 2020-03-05 18:03:04 发布 · 4.6k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python #leetcode

python 同时被 2 个专栏收录

119 篇文章

订阅专栏

leetcode

83 篇文章

订阅专栏

本文介绍了一种将糖果公平分配给一群孩子的算法实现。使用Python编程语言，通过不断递增的糖果数量进行分配，直到糖果用完。算法确保了每个人都能轮流得到糖果，并在糖果不足时按剩余数量分配。

返回的是一个固定长度的列表，在python中，无法通过像java的初始化数组来实现，那么就必须赋值

ans = [0] * num_people

然后定义一个变量give表示每次给的糖果数，这个数每次都会+1

把give和总人数取余，就是索引，因为要不停的遍历每个孩子，假设4个孩子，索引就是0，1，2，3，0，1，2，3，......

class Solution:
    def distributeCandies(self, candies: int, num_people: int):
        ans = [0] * num_people
        give = 0
        while candies > 0:
            ans[give % num_people] += min(give + 1, candies)
            give += 1
            candies -= give
        return ans


if __name__ == "__main__":
    print(Solution().distributeCandies(7, 4))