排排坐,分糖果。

我们买了一些糖果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

分析过程

循环判断糖果candies的数量是否大于0。

定义每次要分的糖果为n,每次循环时,n自增1。

定义结果数组的索引为index,每次循环时,如果没有到最后位置,索引自增1,如果到了最后位置,索引回到位置0。

若大于0,结束循环,返回结果。

若不大于0,进行分糖果的操作。

分糖果时,用糖果candies减去每次要分的糖果n得到剩下的糖果remaining。

若剩下的糖果remaining大于0,那么糖果数量还足够,分n个糖果给结果数组的当前元素,糖果candies更新为剩下的糖果remaining。

若剩下的糖果remaining小于等于0,那么糖果不够用或者刚好用完,把糖果candies全部分给结果数组的当前元素,糖果candies更新为0,结束循环,返回结果。

解答代码

class Solution {
    public int[] distributeCandies(int candies, int num_people) {
        // 定义结果数组results
        int[] results = new int[num_people];

        // 定义结果数组索引
        int index = 0;
        // 定义每次要分糖果
        int n = 1;

        // 循环构造结果数组results,直到糖果candies小于等于0,结束循环
        while (candies > 0) {
            // 获取糖果减去每次要分的糖果
            int remaining = candies - n;

            if (remaining > 0) {
                // 若糖果够减,当前索引的结果数组元素累加每次要分的糖果
                results[index] += n;
                // 糖果更新为减去后的结果
                candies = remaining;
            } else {
                // 若糖果不够减,当前索引的结果数组元素累加剩下的糖果
                results[index] += candies;
                // 糖果设置为0
                candies = 0;
            }

            // 每次要分的糖果自增1
            ++n;
            
            // 结果数组索引往后移动一位,如果到了最后位置,重新回到刚开始位置
            index = index < num_people - 1 ? ++index : 0;
        }
        
        // 返回结果数组
        return results;
    }
}
复制代码

提交结果

执行用时2ms,时间击败34.02%的用户,内存消耗36.1MB,空间击败52.71%的用户。

运行结果