算法分糖果-(贪心)

最新推荐文章于 2025-12-06 22:20:34 发布

原创最新推荐文章于 2025-12-06 22:20:34 发布 · 427 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#分糖果 #贪心算法

算法刷题同时被 2 个专栏收录

41 篇文章

订阅专栏

牛客网

39 篇文章

订阅专栏

文章介绍了如何在满足每个孩子至少一个糖果且相邻孩子中得分高的至少多一些的条件下，计算最少需要多少糖果。通过两次遍历数组，调整`candy`数组来实现这一目标。

牛客网: BM95

题目: 每个孩子至少一个糖果，相邻孩子中得分高的至少多一些，最少多少糖果。

思路: 初始化candy数组元素全为1，从1往右遍历，arr[i]>arr[i-1], 需要candy[i]=candy[i-1]+1；total = candy[n-1], 从右往左遍历，arr[i-1]>arr[i], candy[i-1]<=candy[i]则更新candy[i-1]=candy[i]+1，同时累加到total。

代码:

// go

package main
// import "fmt"

/**
 * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可
 *
 * pick candy
 * @param arr int整型一维数组 the array
 * @return int整型
*/
func candy( arr []int ) int {
    // write code here
    candy := make([]int, len(arr))

    for i := 0; i < len(arr); i++ {
        candy[i] = 1
    }

    for i := 1; i < len(arr); i++ {
        if arr[i] > arr[i-1] {
            candy[i] = candy[i-1] + 1
        }
    }

    total := candy[len(candy)-1]
    for i := len(candy)-1; i > 0; i-- {
        if arr[i-1] > arr[i] && candy[i-1] <= candy[i] {
            candy[i-1] = candy[i] + 1
        }
        total += candy[i-1]
    }
    return total

}