如何用leetcode分糖果

本篇内容介绍了“如何用leetcode分糖果”的有关知识,在实际案例的操作过程中,不少人都会遇到这样的困境,接下来就让小编带领大家学习一下如何处理这些情况吧!希望大家仔细阅读,能够学有所成!

成都创新互联专注于企业成都营销网站建设、网站重做改版、越秀网站定制设计、自适应品牌网站建设、HTML5商城建设、集团公司官网建设、成都外贸网站制作、高端网站制作、响应式网页设计等建站业务,价格优惠性价比高,为越秀等各大城市提供网站开发制作服务。

一、题目内容

排排坐,分糖果。

我们买了一些糖果 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.如果剩下的糖果可以满足应该给的糖果数,就加上满足的糖果数,否则加上剩下的糖果数;

2.如果剩下的糖果数为负,则直接返回;

三、代码

class Solution:
    def distributeCandies(self, candies: int, num_people: int) -> list:
        nums = [0 for _ in range(num_people)]
        j = 1
        while candies:
            for i in range(len(nums)):
                if candies > j:
                    nums[i] += j
                else:
                    nums[i] += candies
                candies -= j
                if candies < 0:
                    return nums
                j += 1
        return nums


if __name__ == '__main__':
    candies = 10
    num_people = 3
    s = Solution()
    ans = s.distributeCandies(candies, num_people)
    print(ans)

“如何用leetcode分糖果”的内容就介绍到这里了,感谢大家的阅读。如果想了解更多行业相关的知识可以关注创新互联网站,小编将为大家输出更多高质量的实用文章!


分享题目:如何用leetcode分糖果
链接URL:http://hbruida.cn/article/iedpoe.html