Python分糖果问题

以下是Python实现的基于优先等级分配糖果问题算法：

def distribute_candy(ratings):
    n = len(ratings)
    candy = [1] * n
    for i in range(1, n):
        if ratings[i] > ratings[i-1]:
            candy[i] = candy[i-1] + 1
    for i in range(n-2, -1, -1):
        if ratings[i] > ratings[i+1]:
            candy[i] = max(candy[i], candy[i+1]+1)
    return sum(candy)

这个算法的思路是，首先每个孩子都先分配一个糖果。然后从左到右遍历一遍，如果右边的孩子比左边的孩子评分高，那么右边的孩子的糖果数就比左边的孩子多一个。接着从右到左再遍历一遍，如果左边的孩子比右边的孩子评分高，那么左边的孩子的糖果数就比右边的孩子多一个。最后将每个孩子手中的糖果数相加即可。