[LeetCode] Candy

最新推荐文章于 2023-02-28 14:31:32 发布

bobyeoh

最新推荐文章于 2023-02-28 14:31:32 发布

阅读量234

点赞数

CC 4.0 BY-SA版权

分类专栏： LeetCode 文章标签：贪心

本文链接：https://blog.youkuaiyun.com/bobyeoh/article/details/70745715

LeetCode 专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种基于贪心策略解决糖果分配问题的方法。通过两次遍历：先正向确保每个评分更高的孩子至少比邻居多一颗糖果，再反向调整以满足条件，最后返回所需糖果总数。

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

贪心策略

1. 正向遍历一遍ratings数组，如果后边邻居的rating更高，则邻居的糖果数+1；

2. 反向遍历一遍ratings数组，如果前边邻居的rating更高，则邻居的糖果数取当前儿童糖果树+1和前边邻居糖果数中的较大值。

3. 所有糖果数量求和，返回结果。

int candy(vector<int>& ratings) {
        int size = ratings.size();
        if (size <= 1) {
            return size;
        }
        vector<int> num(size, 1);
        for (int i = 1; i < size; i++) {
            if (ratings[i] > ratings[i-1]) {
                num[i] = num[i-1] + 1;
            }
        }
        for (int i = size-1; i > 0; i--) {
            if (ratings[i-1] > ratings[i]) {
                num[i-1] = max(num[i]+1, num[i-1]);
            }
        }
        int res = 0;
        for (int i = 0; i < size; i++) {
            res += num[i];
        }
        return res;
}