python算法每日练习day2

今天的是leetcode第1103题原链接在这

分糖果
例：
输入：candies = 7, num_people = 4
输出：[1,2,3,1]
解释：
第一次，ans[0] += 1，数组变为 [1,0,0,0]。
第二次，ans[1] += 2，数组变为 [1,2,0,0]。
第三次，ans[2] += 3，数组变为 [1,2,3,0]。
第四次，ans[3] += 1（因为此时只剩下 1 颗糖果），最终数组变为 [1,2,3,1]

#就按题意循环就完了
class Solution:
    def distributeCandies(self, candies: int, num_people: int) -> List[int]:
        l = [0] * num_people
        
        x = 0
        y = 1
        while candies > 0:
            if x > num_people - 1:
                x = 0

            if candies < y:
                l[x] += candies
            else:
                l[x] += y
            
            x += 1
            candies -= y
            y += 1

        return l