问题描述:
老师想给孩子们分发糖果,有 N
个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。
你需要按照以下要求,帮助老师给这些孩子分发糖果:
(1) 每个孩子至少分配到1
个糖果。
(2) 评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果
。
那么这样下来,老师至少需要准备多少颗糖果呢?
示例:
输入:[1,0,2]
输出:5
解释:你可以分别给这三个孩子分发 2、1、2 颗糖果。
问题分析:
题目解法有多种,题目提示为 贪心和数组,先介绍双数组法。根据题目要求(2)可知评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果
。
(1) 我们可以先从左边向右边扫描
,先满足每个孩子左边的条件
(2) 同理,在从右边向左边扫描
,满足每个孩子右边的条件
(3) 最后根据取两者最大值
即可满足要求
Python3实现:
# @Time :2019/06/17
# @Author :Liu
# 双数组左右扫描法
class Solution:
def candy(self, ratings):
"""
:type ratings: List[int]
:rtype: int
"""
res = [1]*len(ratings)
lbase, rbase = 1, 1
for i in range(1, len(ratings)): # 从左向右扫描
lbase = lbase + 1 if ratings[i] > ratings[i-1] else 1
res[i] = lbase
for i in range(len(ratings)-2, -1, -1): # 从右向左扫描
rbase = rbase + 1 if ratings[i] > ratings[i+1] else 1
res[i] = max(rbase, res[i])
return sum(res)
if __name__ == '__main__':
ratings = [1, 2, 2]
print(Solution().candy(ratings))
一遍扫描方法
这个考虑峰值和谷点,稍微有点难理解,感兴趣的可以看下,下面是其他同学提交的优秀代码可以参考:
# 他人提交的优秀代码段
class Solution:
def candy(self, ratings):
"""
:type ratings: List[int]
:rtype: int
"""
if len(ratings) <= 1: return len(ratings)
candies = 1
up = down = peak = 0
for i in range(1, len(ratings)): # 上升阶段
if ratings[i - 1] < ratings[i]:
up += 1
peak = up
candies += 1 + up
down = 0
elif ratings[i - 1] > ratings[i]: # 下降阶段
down += 1
candies += 1 + down - (1 if peak >= down else 0)
up = 0
else: # 如果相等
peak = up = down = 0
candies += 1
return candies
if __name__ == '__main__':
ratings = [1, 2, 2]
ratings = [4, 5, 6, 5, 4, 3, 2, 1]
print(Solution().candy(ratings))
声明: 总结学习,有问题或不当之处,可以批评指正哦,谢谢。