leetcode刷题题解——135. 分发糖果

分发糖果

题目

n 个孩子站成一排。给你一个整数数组 ratings 表示每个孩子的评分。

你需要按照以下要求，给这些孩子分发糖果：

• 每个孩子至少分配到 1 个糖果。
• 相邻两个孩子评分更高的孩子会获得更多的糖果。

请你给每个孩子分发糖果，计算并返回需要准备的 最少糖果数目 。

动态规划

我采用的是动态规划的方法，设dp[i]表示前i+1个分数时的最优策略
有如下三种情况：

• 如果第i个分数大于第i-1个分数，自然：dp[i] = dp[i-1] + 1
• 如果第i个分数小于第i-1个分数，并且dp[i-1]=1，那么需要将dp[i-1]++,并且i-1前面对dp[i-1]有影响的值，都要进行自增更新，dp[i]=1
• 如果上述条件均不满足：dp[i]=1

最后求和即可

Java代码实现

class Solution {
    public int candy(int[] ratings) {
        int len = ratings.length;
        int[] dp = new int[len];
        dp[0] = 1;
        for (int i=1;i<len;i++){
            if (ratings[i]>ratings[i-1]) dp[i] = dp[i-1] + 1;
            else {
                if (ratings[i]<ratings[i-1]&&dp[i-1]==1) plus(ratings,dp,i-1);
                dp[i] = 1;
            }
        }
        int sum = 0;
        for (int c:dp) sum+=c;
        return sum;

    }

    private void plus(int[] ratings,int[] dp, int i) {
        dp[i]++;
        while (i>0&&ratings[i]<ratings[i-1]&&dp[i-1]-dp[i]==0){
            dp[i-1]++;
            i--;
        }
    }
}

贪心

leetcode上比较好的题解：

1. 先对candy数据均用1填充
2. 遍历ratings数组，如果右边的值大于左边的值，candy[i] = candy[i-1]+1;
3. 再遍历ratings数组，如果左边的值大于右边的值，取candy[i] = max(candy[i + 1] + 1, candy[i])
4. 再对candy数组求和即可

时间复杂度仅为O(n)

Java代码实现

class Solution {
    public int candy(int[] ratings) {
        int len = ratings.length;
        int[] candy = new int[len];
        Arrays.fill(candy, 1);
        for (int i = 1; i < len; i++) {
           if (ratings[i] > ratings[i - 1]){
               candy[i] = candy[i - 1] + 1;
           }
        }
        for (int i = len - 2; i >= 0; i--) {
            if (ratings[i] > ratings[i + 1]){
                candy[i] = Math.max(candy[i + 1] + 1,  candy[i]);
            }
        }
        int result = 0;
        for (int a : candy){
            result += a;
        }
        return result;
    }
}