LeetCode Candy Greedy

最新推荐文章于 2021-07-26 04:49:30 发布

原创最新推荐文章于 2021-07-26 04:49:30 发布 · 406 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 专栏收录该内容

234 篇文章

订阅专栏

本文介绍了一个经典的贪心算法问题——糖果分配问题。该问题的目标是在满足一定条件下尽可能公平地分配最少数量的糖果。文章详细展示了使用两次遍历（从左到右和从右到左）来解决这一问题的具体实现过程。

思路：贪心。

Each child must have at least one candy.
刚开始，每个孩子都得一个candy。
Children with a higher rating get more candies than their neighbors.
从左往右扫描一遍，处理右孩子比左孩子等级高的情况；
再从右往左扫描一遍，处理左孩子比右孩子等级高且左孩子比右孩子candy数少的情况；

class Solution {
public:
    int candy(vector<int>& ratings) {
        int len = ratings.size();
        if(len == 0) {
            return 0;
        }
        if(len == 1) {
            return 1;
        }
        int min_sum = 0;
        vector<int> res(len, 1);
        //from left to right
        for(int i = 0; i <= len - 2; ++i) {
            if(ratings[i + 1] > ratings[i]) {
                res[i + 1] = res[i] + 1;
            }
        }
        //from right to left
        for(int i = len - 1; i >= 1; --i) {
            if(ratings[i - 1] > ratings[i] && res[i - 1] <= res[i]) {
                res[i - 1] = res[i] + 1;
            }
        }
        //sum
        for(int i = 0; i < len; ++i) {
            min_sum += res[i];
        }
        return min_sum;
    }
};

java code:

public class Solution {
    public int candy(int[] ratings) {
        int len = ratings.length;
        int[] res = new int[len];
        Arrays.fill(res, 1);

        // from left to right
        for(int i = 0; i <= len - 2; ++i) {
            if(ratings[i + 1] > ratings[i]) {
                res[i + 1] = res[i] + 1;
            }
        }

        // from right to left
        for(int i = len - 1; i >= 1; --i) {
            if(ratings[i - 1] > ratings[i] && res[i - 1] <= res[i]) {
                res[i - 1] = res[i] + 1;
            }
        }

        int sum = 0;
        for(int i = 0; i < len; ++i) {
            sum += res[i];
        }

        return sum;
    }
}