博主将 LeetCode135 和 LeetCode1103 两道题放在一起，虽然都是分糖果，但是思路却完全不一样。仅供借鉴。
目录
一、题目
LeetCode135：分发糖果
老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。
你需要按照以下要求，帮助老师给这些孩子分发糖果：
每个孩子至少分配到 1 个糖果。
评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。
那么这样下来，老师至少需要准备多少颗糖果呢？
LeetCode1103：分糖果II
排排坐，分糖果。
我们买了一些糖果 candies，打算把它们分给排好队的 n = num_people 个小朋友。
给第一个小朋友 1 颗糖果，第二个小朋友 2 颗，依此类推，直到给最后一个小朋友 n 颗糖果。
然后，我们再回到队伍的起点，给第一个小朋友 n + 1 颗糖果，第二个小朋友 n + 2 颗，依此类推，直到给最后一个小朋友 2 * n 颗糖果。
重复上述过程（每次都比上一次多给出一颗糖果，当到达队伍终点后再次从队伍起点开始），直到我们分完所有的糖果。注意，就算我们手中的剩下糖果数不够（不比前一次发出的糖果多），这些糖果也会全部发给当前的小朋友。
返回一个长度为 num_people、元素之和为 candies 的数组，以表示糖果的最终分发情况（即 ans[i] 表示第 i 个小朋友分到的糖果数）。
二、示例
LeetCode135：分发糖果
示例 1：
输入：[1,0,2]
输出：5
解释：你可以分别给这三个孩子分发 2、1、2 颗糖果。
示例 2：
输入：[1,2,2]
输出：4
解释：你可以分别给这三个孩子分发 1、2、1 颗糖果。
第三个孩子只得到 1 颗糖果，这已满足上述两个条件。
LeetCode1103：分糖果II
示例 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
三、思路
LeetCode135：分发糖果
方案一：
1、每个人先发一个糖果
2、遍历————
- 如果当前这个人比左边的人分数高，且他的糖果数没有左边的人多，则糖果数为左边糖果数 +1
- 如果当前这个人比右边的人分数高，且他的糖果数没有右边的人多，则糖果数为右边糖果数 +1
3、直到状态不发生变化，返回总糖果数即可。注：该方法是正确的，适用于用例较少的情况。用例太大会超出时间限制！
只是因为LeetCode上的测试用例太太太太大。博主只是截了一小部分，可见用例有多大。
方案二：
该方法通过LeetCode!
1、每个人先发一个糖果
2、遍历2次————
第一次遍历，考虑每个孩子左边的孩子
到二次遍历，考虑每个孩子右边的孩子
LeetCode1103：分糖果II
while 循环，直到糖果 candies 分完为止。
按照 1，2，3，...，n 的糖果分发给小朋友。num = 1，2，3， ... ， n
每次分发前判断糖果数是否大于分发的糖果数num：
- 若大于，则继续分发；
- 若小于，则将剩下的糖果数分发给当前的小朋友；
- 若糖果分发完，直接返回。
四、代码
LeetCode135：分发糖果
1、方法一
class Solution:
def candy(self, ratings) -> int:
"""
:param ratings: List[int]
:return: int
"""
"""
思路：
1、每个人先发一个糖果
2、遍历————
- 如果当前这个人比左边的人分数高，且他的糖果数没有左边的人多，则糖果数为左边糖果数 +1
- 如果当前这个人比右边的人分数高，且他的糖果数没有右边的人多，则糖果数为右边糖果数 +1
3、直到状态不发生变化，返回总糖果数即可。
"""
n = len(ratings)
if n == 0:
return 0
elif n == 1:
return 1
candy = [1 for _ in range(n)]
while True:
flag = True
for i in range(n):
if i > 0 and ratings[i - 1] > ratings[i] and candy[i] >= candy[i - 1]:
candy[i - 1] = candy[i] + 1
flag = False
if i < n - 1 and ratings[i + 1] > ratings[i] and candy[i] >= candy[i + 1]:
candy[i + 1] = candy[i] + 1
flag = False
if flag:
break
# print(candy)
return sum(candy)
if __name__ == '__main__':
# test = [1, 0, 2]
# test = [1,3,2,2,1]
test = [1,2,87,87,87,2,1]
s = Solution()
ans = s.candy(test)
print(ans)2、方法二
class Solution:
def candy(self, ratings) -> int:
"""
:param ratings: List[int]
:return: int
"""
"""
优化思路1：
1、每个人先发一个糖果
2、遍历2次————
第一次遍历，考虑每个孩子左边的孩子
到二次遍历，考虑每个孩子右边的孩子
"""
n = len(ratings)
candy = [1 for _ in range(n)]
# 每个孩子的左边
for i in range(1, n):
if ratings[i] > ratings[i - 1] and candy[i] <= candy[i - 1]:
candy[i] = candy[i - 1] + 1
# 每个孩子的右边
for i in range(n - 2, -1, -1):
if ratings[i] > ratings[i + 1] and candy[i] <= candy[i + 1]:
candy[i] = candy[i + 1] + 1
return sum(candy)
if __name__ == '__main__':
# test = [1, 0, 2]
# test = [1,3,2,2,1]
test = [1,2,87,87,87,2,1]
s = Solution()
ans = s.candy(test)
print(ans)LeetCode1103：分糖果II
class Solution:
def distributeCandies(self, candies: int, num_people: int):
"""
:param candies: int 7
:param num_people: int 4
:return: List[int]
"""
ans = [0 for _ in range(num_people)]
num = 1
while candies:
for i in range(num_people):
if candies - num >= 0:
ans[i] += num
candies -= num
num += 1
elif candies >= 0:
ans[i] += candies
return ans
else:
return ans
# return ans
if __name__ == '__main__':
andies = 7
num_people = 4
s = Solution()
ans = s.distributeCandies(andies, num_people)
print(ans)