分糖果

一个很有趣的个人博客,不信你来撩 fangzengye.com

class Solution:
    def distributeCandies(self, candies: int, num_people: int) -> List[int]:
        List = [0]*num_people
        flag = 1
        num = 1
        while flag:
            for i in range(num_people):
                if candies>num:
                    candies=candies-num
                    List[i]+=num
                    num+=1


                else:
                    List[i]+=candies
                    flag = 0
                    break
        return List