《leetcode-go》糖果问题

less_cold

于 2020-08-11 23:40:15 发布

阅读量262

点赞数

CC 4.0 BY-SA版权

分类专栏： leetCode-go

本文链接：https://blog.youkuaiyun.com/less_cold/article/details/107947268

leetCode-go 专栏收录该内容

81 篇文章

订阅专栏

本文介绍了一种高效的算法，用于解决在给定评分的情况下，如何最小化分配给每个小朋友糖果的数量，同时确保评分高的小朋友比旁边评分低的小朋友得到更多糖果的问题。通过双向遍历和动态调整糖果数目的策略，该算法能够快速找到最优解。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

有N个小朋友站在一排，每个小朋友都有一个评分

你现在要按以下的规则给孩子们分糖果：

每个小朋友至少要分得一颗糖果
分数高的小朋友要他比旁边得分低的小朋友分得的糖果多

你最少要分发多少颗糖果？

注意，比较糖果数量的时候，等于也不行

package main

/**
  * 
  * @param ratings int整型一维数组 
  * @return int整型
*/
func candy( ratings []int ) int {
    // write code here
    num := len(ratings)
	if num == 0 {
		return 0
	}
	candy := make([]int, num)
	candy[0] = 1
	for i := 1; i < num; i++ {
		if ratings[i] > ratings[i-1] {
			candy[i] = candy[i-1] + 1
		} else {
			candy[i] = 1
		}
	}
	for i := num - 1; i > 0; i-- {
		if ratings[i] < ratings[i-1] && candy[i] >= candy[i-1] {
			candy[i-1] = candy[i] + 1
		}
	}
	var ret int
	for _, cnt := range candy {
		ret += cnt
	}
	return ret
}