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