1103. 分糖果 II 數學問題

https://www.b2bchain.cn/6185.html 

//排排坐，分糖果。 
//
// 我們買了一些糖果 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]。
// 
//
// 
//
// 提示： 
//
// 
// 1 <= candies <= 10^9 
// 1 <= num_people <= 1000 
// 
// Related Topics 數學


//leetcode submit region begin(Prohibit modification and deletion)
class Solution {
    public int[] distributeCandies(int candies, int num_people) {
        int[] ans = new int[num_people];
        int i = 0;
        while (candies != 0) {
            ans[i % num_people] += Math.min(candies, i + 1);
            candies -= Math.min(candies, i + 1);
            i += 1;
        }
        return ans;
    }
}