1104-分糖果 II

lily_wang 發(fā)布于2019-08-16 18:26 / 2639人閱讀

摘要：重復上述過程每次都比上一次多給出一顆糖果，當?shù)竭_隊伍終點后再次從隊伍起點開始，直到我們分完所有的糖果。返回一個長度為元素之和為的數(shù)組，以表示糖果的最終分發(fā)情況即表示第個小朋友分到的糖果數(shù)。

前言

Weekly Contest 143的分糖果 II

排排坐，分糖果。
我們買了一些糖果 candies，打算把它們分給排好隊的 n = num_people 個小朋友。
給第一個小朋友 1 顆糖果，第二個小朋友 2 顆，依此類推，直到給最后一個小朋友 n 顆糖果。
然后，我們再回到隊伍的起點，給第一個小朋友 n + 1 顆糖果，第二個小朋友 n + 2 顆，依此類推，直到給最后一個小朋友 2 * n 顆糖果。
重復上述過程（每次都比上一次多給出一顆糖果，當?shù)竭_隊伍終點后再次從隊伍起點開始），直到我們分完所有的糖果。注意，就算我們手中的剩下糖果數(shù)不夠（不比前一次發(fā)出的糖果多），這些糖果也會全部發(fā)給當前的小朋友。
返回一個長度為 num_people、元素之和為 candies 的數(shù)組，以表示糖果的最終分發(fā)情況（即 ans[i] 表示第 i 個小朋友分到的糖果數(shù)）。
示例1：
輸入：candies = 7, num_people = 4
輸出：[1,2,3,1]
解釋：
第一次，ans[0] += 1，數(shù)組變?yōu)?[1,0,0,0]。
第二次，ans[1] += 2，數(shù)組變?yōu)?[1,2,0,0]。
第三次，ans[2] += 3，數(shù)組變?yōu)?[1,2,3,0]。
第四次，ans[3] += 1（因為此時只剩下 1 顆糖果），最終數(shù)組變?yōu)?[1,2,3,1]。
示例2：
輸入：candies = 10, num_people = 3
輸出：[5,2,3]
解釋：
第一次，ans[0] += 1，數(shù)組變?yōu)?[1,0,0]。
第二次，ans[1] += 2，數(shù)組變?yōu)?[1,2,0]。
第三次，ans[2] += 3，數(shù)組變?yōu)?[1,2,3]。
第四次，ans[0] += 4，最終數(shù)組變?yōu)?[5,2,3]。
提示：

1 <= candies <= 10^9
1 <= num_people <= 1000

解題思路

本題十分簡單，具體邏輯只要查看實現(xiàn)代碼及配套注釋即可。

實現(xiàn)代碼

   /**
     * 1104. 分糖果 II
     * @param candies
     * @param num_people
     * @return
     */
    public int[] distributeCandies(int candies, int num_people) {
        int[] result=new int[num_people];
        // 當前遍歷的索引
        int index=0;
        // 發(fā)放到個人的糖果數(shù)
        int num=1;
        while (candies>0){
            // 給小朋友發(fā)糖果
            result[index]+=num;
            // 剩余糖果數(shù)減少
            candies-=num;
            // 是否發(fā)到隊伍最后一位小朋友
            if(indexnum){
                ++num;
            }else{
                num=candies;
            }
        }
        return result;
    }