1103-distribute-candies-to-people.js

/**
 * 1103. Distribute Candies to People
 * https://leetcode.com/problems/distribute-candies-to-people/
 * Difficulty: 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.
 */

/**
 * @param {number} candies
 * @param {number} num_people
 * @return {number[]}
 */
var distributeCandies = function(candies, num_people) {
  const answer = new Array(num_people).fill(0);

  for (let amount = 0; candies > 0; candies -= amount) {
    answer[amount % num_people] += Math.min(candies, ++amount);
  }

  return answer;
};