Welcome to Subscribe On Youtube
Formatted question description: https://leetcode.ca/all/1103.html
1103. Distribute Candies to People (Easy)
We distribute some number of candies
, to a row of n = num_people
people in the following way:
We then give 1 candy to the first person, 2 candies to the second person, and so on until we give n
candies to the last person.
Then, we go back to the start of the row, giving n + 1
candies to the first person, n + 2
candies to the second person, and so on until we give 2 * n
candies to the last person.
This process repeats (with us giving one more candy each time, and moving to the start of the row after we reach the end) until we run out of candies. The last person will receive all of our remaining candies (not necessarily one more than the previous gift).
Return an array (of length num_people
and sum candies
) that represents the final distribution of candies.
Example 1:
Input: candies = 7, num_people = 4 Output: [1,2,3,1] Explanation: On the first turn, ans[0] += 1, and the array is [1,0,0,0]. On the second turn, ans[1] += 2, and the array is [1,2,0,0]. On the third turn, ans[2] += 3, and the array is [1,2,3,0]. On the fourth turn, ans[3] += 1 (because there is only one candy left), and the final array is [1,2,3,1].
Example 2:
Input: candies = 10, num_people = 3 Output: [5,2,3] Explanation: On the first turn, ans[0] += 1, and the array is [1,0,0]. On the second turn, ans[1] += 2, and the array is [1,2,0]. On the third turn, ans[2] += 3, and the array is [1,2,3]. On the fourth turn, ans[0] += 4, and the final array is [5,2,3].
Constraints:
 1 <= candies <= 10^9
 1 <= num_people <= 1000
Related Topics:
Math
Solution 1.

class Solution { public int[] distributeCandies(int candies, int num_people) { int[] array = new int[num_people]; int index = 0; int curCandies = 1; while (candies > 0) { int num = Math.min(curCandies, candies); array[index] += num; candies = num; curCandies++; index = (index + 1) % num_people; } return array; } } ############ class Solution { public int[] distributeCandies(int candies, int num_people) { int[] ans = new int[num_people]; for (int i = 0; candies > 0; ++i) { ans[i % num_people] += Math.min(candies, i + 1); candies = Math.min(candies, i + 1); } return ans; } }

// OJ: https://leetcode.com/problems/distributecandiestopeople/ // Time: O(sqrt(candies)) // Space: O(1) class Solution { public: vector<int> distributeCandies(int candies, int n) { vector<int> ans(n); for (int c = 1, i = 0; candies; i = (i + 1) % n, ++c) { int val = min(c, candies); ans[i] += val; candies = val; } return ans; } };

class Solution: def distributeCandies(self, candies: int, num_people: int) > List[int]: ans = [0] * num_people i = 0 while candies > 0: ans[i % num_people] += min(candies, i + 1) candies = min(candies, i + 1) i += 1 return ans ############ # 1103. Distribute Candies to People # https://leetcode.com/problems/distributecandiestopeople/ class Solution: def distributeCandies(self, candies: int, n: int) > List[int]: t = i = 0 res = [0] * n while candies > 0: t += 1 give = min(t, candies) candies = give res[i] += give i = (i + 1) % n return res

func distributeCandies(candies int, num_people int) []int { ans := make([]int, num_people) for i := 0; candies > 0; i++ { ans[i%num_people] += min(candies, i+1) candies = min(candies, i+1) } return ans } func min(a, b int) int { if a < b { return a } return b }

function distributeCandies(candies: number, num_people: number): number[] { const ans: number[] = Array(num_people).fill(0); for (let i = 0; candies > 0; ++i) { ans[i % num_people] += Math.min(candies, i + 1); candies = Math.min(candies, i + 1); } return ans; }