LeetCode刷题-分糖果II

前言说明

算法学习，日常刷题记录。

题目连接

分糖果II

题目内容

排排坐，分糖果。

我们买了一些糖果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%的用户。

原文链接

原文链接：分糖果II