如何用leetcode分糖果

本篇內(nèi)容介紹了“如何用leetcode分糖果”的有關(guān)知識，在實際案例的操作過程中，不少人都會遇到這樣的困境，接下來就讓小編帶領(lǐng)大家學(xué)習(xí)一下如何處理這些情況吧！希望大家仔細(xì)閱讀，能夠?qū)W有所成！

一、題目內(nèi)容

排排坐，分糖果。

我們買了一些糖果 candies，打算把它們分給排好隊的 n = num_people 個小朋友。

給第一個小朋友 1 顆糖果，第二個小朋友 2 顆，依此類推，直到給最后一個小朋友 n 顆糖果。

然后，我們再回到隊伍的起點，給第一個小朋友 n + 1 顆糖果，第二個小朋友 n + 2 顆，依此類推，直到給最后一個小朋友 2 * n 顆糖果。

重復(fù)上述過程（每次都比上一次多給出一顆糖果，當(dāng)?shù)竭_隊伍終點后再次從隊伍起點開始），直到我們分完所有的糖果。注意，就算我們手中的剩下糖果數(shù)不夠（不比前一次發(fā)出的糖果多），這些糖果也會全部發(fā)給當(dāng)前的小朋友。

返回一個長度為 num_people、元素之和為 candies 的數(shù)組，以表示糖果的最終分發(fā)情況（即 ans[i] 表示第 i 個小朋友分到的糖果數(shù)）。

示例 1：

輸入：candies = 7, num_people = 4
輸出：[1,2,3,1]
解釋：
第一次，ans[0] += 1，數(shù)組變?yōu)?[1,0,0,0]。
第二次，ans[1] += 2，數(shù)組變?yōu)?[1,2,0,0]。
第三次，ans[2] += 3，數(shù)組變?yōu)?[1,2,3,0]。
第四次，ans[3] += 1（因為此時只剩下 1 顆糖果），最終數(shù)組變?yōu)?[1,2,3,1]。

示例 2：

輸入：candies = 10, num_people = 3
輸出：[5,2,3]
解釋：
第一次，ans[0] += 1，數(shù)組變?yōu)?[1,0,0]。
第二次，ans[1] += 2，數(shù)組變?yōu)?[1,2,0]。
第三次，ans[2] += 3，數(shù)組變?yōu)?[1,2,3]。
第四次，ans[0] += 4，最終數(shù)組變?yōu)?[5,2,3]。

提示：

1 <= candies <= 10^9
1 <= num_people <= 1000

二、解題思路

沒啥說的。直接遍歷：

1.如果剩下的糖果可以滿足應(yīng)該給的糖果數(shù)，就加上滿足的糖果數(shù)，否則加上剩下的糖果數(shù)；

2.如果剩下的糖果數(shù)為負(fù)，則直接返回；

三、代碼

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分糖果”的內(nèi)容就介紹到這里了，感謝大家的閱讀。如果想了解更多行業(yè)相關(guān)的知識可以關(guān)注億速云網(wǎng)站，小編將為大家輸出更多高質(zhì)量的實用文章！