Leetcode 1103. Distribute Candies to People

Tyan

发布于 2022-08-11 15:03:01

1760

发布于 2022-08-11 15:03:01

文章被收录于专栏：SnailTyanSnailTyan

文章作者：Tyan 博客：noahsnail.com | CSDN | 简书

1. Description

2. Solution

**解析：**Version 1，循环遍历即可，current为当前这次要分配的糖果数量，每次要分配的糖果数量加1，如果不够，则剩余糖果分配给最后一个。

Version 1

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

**解析：**Version 2，index可以用current替代。

Version 2

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