题目:
老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
相邻的孩子中,评分高的孩子必须获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
示例 1:
输入: [1,0,2]
输出: 5
解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
示例 2:
输入: [1,2,2]
输出: 4
解释: 你可以分别给这三个孩子分发 1、2、1 颗糖果。第三个孩子只得到 1 颗糖果,这已满足上述两个条件。
来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/candy
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
解题思路:
规则定义: 设学生 A 和学生 B 左右相邻,A 在 BB左边;
左规则: 当 ratings_B>ratings_A时,B的糖比 A的糖数量多。
右规则: 当 ratings_A>ratings_B时,A 的糖比 B 的糖数量多。
相邻的学生中,评分高的学生必须获得更多的糖果 等价于 所有学生满足左规则且满足右规则。
算法流程:
1、创建左右规则的两个发糖列表,数值全部预设为1;
2、利用左规则正向遍历一次,当 ratings_B>ratings_A时,B的糖比 A 的糖数量多1;
3、再利用右规则反向遍历一次,当 ratings_A>ratings_B时,A 的糖比 B 的糖数量多1;
4、最后取同时满足左右规则的数值即可(即左右规则列表中每个索引的较大者)。
代码(python):
class Solution: #贪心算法,左右规则遍历,取最大值
def candy(self, ratings: List[int]) -> int:
length = len(ratings)
left = [1 for _ in range(length)] #先将每个人的糖都设为1
right = left[:]
for i in range(1, length): #正向遍历
if ratings[i]>ratings[i-1]: #左规则
left[i] = left[i-1] + 1
count = left[-1] #创建保存较大值的变量
for i in range(length-2, -1, -1): #反向遍历
if ratings[i]>ratings[i+1]: #右规则
right[i] = right[i+1] + 1
count += max(left[i], right[i]) #取同时满足左右规则的发糖数
#print(left)
#print(right)
return count