老師想給孩子們分發糖果,有 N 個孩子站成了一條直線,老師會根據每一個孩子的表現,預先給他們評分。 你須要按照如下要求,幫助老師給這些孩子分發糖果: 每一個孩子至少分配到 1 個糖果。 相鄰的孩子中,評分

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;
    }
}
相關文章
相關標籤/搜索