返回的是一个固定长度的列表,在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))