1103. 分糖果 II Easy

排排坐,分糖果。

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

题目大意:按顺序循环递增地发放糖果,计算每个人分到的糖果数。

分析:

(1)根据糖果数和人数可以计算所有人都能分到糖果次数的turn、能分到turn+1次糖果的人数rem、最后一次比前一次发的少的糖果数rest。

(2)序号i<=rem的学生可以分到turn+1次糖果,其所得到的糖果总数=i+(n+i)+(2*n+i)+...+(turn*n+i)=(n+2*n+..+turn*n)+(turn+1)*i=turn*(turn+1)*n/2+(turn+1)*i;

(3)``序号i>rem的学生可以分到turn次糖果,``其所得到的糖果总数=i+(n+i)+(2*n+i)+...+((turn-1)*n+i)=(n+2*n+..+(turn-1)*n)+turn*i=(turn-1)*turn+*n/2+turn*i;

(4)rest个糖果会分给序号为rem+1的学生,因此第rem+1个学生得到的糖果数需再增加rest。

cpp 复制代码
class Solution {
public:
    vector<int> distributeCandies(int candies, int num_people) {
        vector<int> ans(num_people);
        int x=round(sqrt(candies*2+1))-1;
        int turn=x/num_people,rem=x%num_people,rest=candies-(1+x)*x/2;
        for(int i=1;i<=rem;++i){
            ans[i-1]=(1+turn)*turn*num_people/2+(turn+1)*i;
        }
        for(int i=rem+1;i<=num_people;++i){
            ans[i-1]=turn*(turn-1)*num_people/2+turn*i;
        }
        ans[rem]+=rest;
        return ans;
    }
};
相关推荐
虾球xz14 分钟前
游戏引擎学习第244天: 完成异步纹理下载
c++·学习·游戏引擎
矛取矛求21 分钟前
C++区别于C语言的提升用法(万字总结)
c语言·c++
mit6.82422 分钟前
[贪心_7] 最优除法 | 跳跃游戏 II | 加油站
数据结构·算法·leetcode
keep intensify24 分钟前
通讯录完善版本(详细讲解+源码)
c语言·开发语言·数据结构·算法
ephemerals__27 分钟前
【c++11】c++11新特性(下)(可变参数模板、default和delete、容器新设定、包装器)
开发语言·c++
shix .30 分钟前
2025年PTA天梯赛正式赛 | 算法竞赛,题目详解
数据结构·算法
风铃儿~39 分钟前
Java面试高频问题(26-28)
java·算法·面试
wuqingshun31415941 分钟前
蓝桥杯 4. 卡片换位
算法·职场和发展·蓝桥杯
江沉晚呤时43 分钟前
深入了解C# List集合及两种常见排序算法:插入排序与堆排序
windows·sql·算法·oracle·c#·排序算法·mybatis
egoist20231 小时前
【C++指南】告别C字符串陷阱:如何实现封装string?
开发语言·数据结构·c++·c++11·string·auto·深/浅拷贝