题目:
老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。你需要按照以下要求,帮助老师给这些孩子分发糖果:
每个孩子至少分配到 1 个糖果。
评分更高的孩子必须比他两侧的邻位孩子获得更多的糖果。
那么这样下来,老师至少需要准备多少颗糖果呢?
思路:
将每个数当看作一个波浪线中的峰值,从左向右扫描一次,再从右向左扫描一次,扫描到当前数即停止,取两者中较大的值。具体过程如下:
扫描两次:
第一次,从左向右扫描,将第一个人拥有的糖果数设定为1,即left[0]=1。扫描时,若当前值大于前一个值,则将当前值在前一个值的基础上加1,否则令当前值为1。
第二次,从右向左扫描,边扫描边判断,并对最终结果进行累加。将第一个人拥有的糖果数设定为1,即right[n-1]=0。扫描时,若当前值大于前一个值,则将当前值在前一个值的基础上加1,否则令当前值为1。取left[i]和right[i]中较大的数,对最终结果进行累加。
解答:
class Solution:
def candy(self, ratings: List[int]) -> int:
n = len(ratings)
left = [0] * n
for i in range(n):
if i > 0 and ratings[i] > ratings[i - 1]:
left[i] = left[i - 1] + 1
else:
left[i] = 1
right = ret = 0
for i in range(n - 1, -1, -1):
if i < n - 1 and ratings[i] > ratings[i + 1]:
right += 1
else:
right = 1
ret += max(left[i], right)
return ret
方法二:仅用一个数组
class Solution:
def candy(self, ratings: List[int]) -> int:
n=len(ratings)
res=[1]*n
#一次是从左到右遍历,只比较右边孩子评分比左边大的情况。
for i in range(1,n):
if ratings[i]>ratings[i-1]:
res[i]=res[i-1]+1
ans=res[-1]
#一次是从右到左遍历,只比较左边孩子评分比右边大的情况。
for i in range(n-2,-1,-1):
if ratings[i]>ratings[i+1]:
res[i]=max(res[i],res[i+1]+1)
ans+=res[i]
return ans