每日一题: 分糖果

排排坐，分糖果。

我们买了一些糖果 candies，打算把它们分给排好队的 n = num_people 个小朋友。

给第一个小朋友 1 颗糖果，第二个小朋友 2 颗，依此类推，直到给最后一个小朋友 n 颗糖果。

然后，我们再回到队伍的起点，给第一个小朋友 n + 1 颗糖果，第二个小朋友 n + 2 颗，依此类推，直到给最后一个小朋友 2 * n 颗糖果。

重复上述过程（每次都比上一次多给出一颗糖果，当到达队伍终点后再次从队伍起点开始），直到我们分完所有的糖果。注意，就算我们手中的剩下糖果数不够（不比前一次发出的糖果多），这些糖果也会全部发给当前的小朋友。

返回一个长度为 num_people、元素之和为 candies 的数组，以表示糖果的最终分发情况（即 ans[i] 表示第 i 个小朋友分到的糖果数）。

示例 1：

输入： candies = 7, num_people = 4
输出： [1,2,3,1]
解释：
第一次，ans[0] += 1，数组变为 [1,0,0,0]。
第二次，ans[1] += 2，数组变为 [1,2,0,0]。
第三次，ans[2] += 3，数组变为 [1,2,3,0]。
第四次，ans[3] += 1（因为此时只剩下 1 颗糖果），最终数组变为 [1,2,3,1]。

示例 2：

输入： candies = 10, num_people = 3
输出： [5,2,3]
解释：
第一次，ans[0] += 1，数组变为 [1,0,0]。
第二次，ans[1] += 2，数组变为 [1,2,0]。
第三次，ans[2] += 3，数组变为 [1,2,3]。
第四次，ans[0] += 4，最终数组变为 [5,2,3]。

代码:

    var distributeCandies = function (candies, num_people) {
      let arr = [] // 设定空数组
      arr.length = num_people // 给数组长度
      arr.fill(0) // 填充数组 [0,0,0,0,...]
      let initNum = 1 // 分发的糖果数量
      let resmainNum = candies // 剩余糖果数量

      let i = 0 // 下标

      while (resmainNum > 0) { // 还剩余有糖果就继续循环
        if (i >= num_people) { // 当数组的下标大于分的人的数量时 将i置为0
          i = 0
        }
        if (resmainNum >= initNum) { // 当剩余糖果比分发的糖果数量多时 给对应下标为i的元素
          arr[i] = arr[i] + initNum
          resmainNum = resmainNum - initNum
          initNum++
        } else { // 剩余糖果比分发的糖果数量少时 全给最后一个人
          arr[i] = arr[i] + resmainNum
          resmainNum = 0
        }
        i++
      }
      return arr
    };

leetCode参考答案

var distributeCandies = function(candies, num_people) {
    const ans = new Array(num_people).fill(0);
    let i = 0;
    while (candies !== 0) {
        ans[i % num_people] += Math.min(candies, i + 1);
        candies -= Math.min(candies, i + 1);
        i++;
    }
    return ans;
};