老师想给孩子们分发糖果,有 N 个孩子站成了一条直线,老师会根据每个孩子的表现,预先给他们评分。 你需要按照以下要求,帮助老师给这些孩子分发糖果: 每个孩子至少分配到 1 个糖果。 相邻的孩子中,评分高的孩子必须获得更多的糖果。 那么这样下来,老师至少需要准备多少颗糖果呢? 来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/candy 著

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

猜你喜欢

转载自www.cnblogs.com/q-1993/p/11323509.html