20240603 1103 - Easy - Array

1103. Distribute Candies to People

class Solution:
    def distributeCandies(self, candies: int, num_people: int) -> List[int]:
        ans = [0 for i in range(num_people)]
        i = 0
        while candies > 0:
            if i + 1 <= candies:
                candies -= i + 1
                ans[i % num_people] += i + 1
            else:
                ans[i % num_people] += candies
                break
            i += 1
        return ans