分糖果 II

// Java Stream Solution
// 思路：
// 1. 初始化一个大小为 num_people 的数组来记录每个人分到的糖果数。
// 2. 用一个变量记录当前要分配的糖果数量，初始化为 1。
// 3. 用一个变量记录当前剩余的糖果数量。
// 4. 使用 Stream 的 iterate 方法循环分配糖果：
//    - 根据剩余糖果数更新每个小朋友分到的糖果。

import java.util.stream.IntStream;

public int[] distributeCandies(int candies, int num_people) {
    int[] result = new int[num_people];
    int[] candyCount = {1}; // 当前分配的糖果数

    IntStream.iterate(0, i -> (i + 1) % num_people)
             .takeWhile(i -> candies > 0)
             .forEach(i -> {
                 if (candies >= candyCount[0]) {
                     result[i] += candyCount[0];
                     candies -= candyCount[0];
                 } else {
                     result[i] += candies;
                     candies = 0;
                 }
                 candyCount[0]++;
             });
    return result;
}