2023.7.5
这个题看完随想录的视频,感觉是妙蛙种子,但不道是为啥。
我思考了一下如此遍历两次如何保证满足题意:
意思是甲>乙时,向右遍历时,乙会得到一个糖果,向左遍历时甲一定比乙多一个糖果,最终的糖果数即为乙得到了n颗,甲得到了至少n+1颗。而甲<乙时同理。甲=乙时其实是无所谓的,无论甲乙拿到多少糖果都是满足题意的。
因此这样遍历两次后,可以保证任意相邻的两个孩子,都是得分高的那个孩子拿到的糖果多。这就是局部最优。
class Solution:
def candy(self, ratings: List[int]) -> int:
n = len(ratings)
keeper = [1] * n # 初始化为1
for i in range(1, n):
if ratings[i] > ratings[i-1]:
keeper[i] = keeper[i-1] + 1
for i in range(n-2, -1, -1):
if ratings[i] > ratings[i+1]:
keeper[i] = max(keeper[i], keeper[i+1]+1)
return sum(keeper)