题目
题解
思路:
1.首先每个人至少一块糖果
2.循环两遍数组,从左往右,右边分高的至少比左边多一块;
从右往左,因为这个时候可能左边分比右边高的同时糖果比右边多,这个时候维持原状,糖果比右边少的话就要比右边多一个
class Solution:
def candy(self, ratings: List[int]) -> int:
n = len(ratings)
nums = [1]*n #每人最少有一块
for i in range(0, n-1): #右边分高的至少比左边多一块
if ratings[i] < ratings[i+1]:
nums[i+1] = nums[i]+1
for i in range(n-1, 0, -1): #左边分高的如果糖果没右边的多,就要分给比右边最少多一块
if ratings[i-1] > ratings[i]:
nums[i-1] = max(nums[i-1], nums[i]+1)
return sum(nums)