贪心:leetcode135(困难)——分发糖果

题目

在这里插入图片描述

题解

思路:
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)

Guess you like

Origin blog.csdn.net/memory_MM_forever/article/details/121564155