有N个孩子站成一排,每个孩子有一个分值。给这些孩子派发糖果,需要满足如下需求:
1、每个孩子至少分到一个糖果
2、分值更高的孩子比他相邻位的孩子获得更多的糖果
求至少需要分发多少糖果?
nums = list(map(int, input().split(',')))
candy = [1 for _ in range(len(nums))]
for i in range(1, len(nums)):
if nums[i] >nums[i-1]:
candy[i] = candy[i-1] +1
for i in range(len(nums)-2, -1, -1):
if nums[i] > nums[i+1]:
candy[i] = max(candy[i], candy[i+1]+1)
print(sum(candy)) """"
从左至右,看自己左侧的分数,若比自己小,在他的基础上加一
从右至左,看自己右侧的分数,若比自己小,max(自己,右侧+1)
"""
if __name__ == "__main__":
a = list(map(int, input().strip().split(',')))
n = len(a)
ans = [1] * n
for i in range(1, n):
if a[i] > a[i - 1]:
ans[i] = ans[i - 1] + 1
for i in range(n - 2, -1, -1):
if a[i] > a[i + 1]:
ans[i] = max(ans[i], ans[i + 1] + 1)
print(sum(ans))