如何用leetcode分糖果

本篇內容介紹了“如何用leetcode分糖果”的有關知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領大家學習一下如何處理這些情況吧！希望大家仔細閱讀，能夠學有所成！

一、題目內容

排排坐，分糖果。

我們買了一些糖果 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分糖果”的內容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業相關的知識可以關注億速云網站，小編將為大家輸出更多高質量的實用文章！