Distribute Candies to People LeetCode solution in Python

1
class Solution:
2
    def distributeCandies(self, candies: int, num_people: int) -> List[int]:
3
        candy_dict = {}
4
        for i in range(num_people):
5
            candy_dict[i] = 0
6

7
        candy, i, totalCandy = 1, 0, 0
8
        while totalCandy < candies:
9
            if i >= num_people:
10
                i = 0
11
            if candies - totalCandy >= candy:
12
                candy_dict[i] += candy
13
                totalCandy += candy
14
            else:
15
                candy_dict[i] += candies - totalCandy
16
                totalCandy += candies - totalCandy
17
            i += 1
18
            candy += 1
19
        return candy_dict.values()