lintcode(412)分糖果

Description:

有 N 个小孩站成一列。每个小孩有一个评级。

按照以下要求，给小孩分糖果：

• 每个小孩至少得到一颗糖果。 

• 评级越高的小孩可以得到更多的糖果。

需最少准备多少糖果？

Explanation:

给定评级 = [1, 2], 返回 3.

给定评级 = [1, 1, 1], 返回 3.

给定评级 = [1, 2, 2], 返回 4. ([1,2,1]).

Solution:

感觉题目描述不清楚，就查看了一些参考，题目描述应该为  相邻两个人，评级越高，获得的糖果越多

所以从前往后，再从后往前，遍历两次，比较相邻两点，并将糖果数记录在dp中。

public class Solution {
    /**
     * @param ratings Children's ratings
     * @return the minimum candies you must give
     */
    public int candy(int[] ratings) {
        // Write your code here
        int len = ratings.length;
        int[] dp = new int[len];
        for(int i = 0;i<len;i++){
            dp[i] = 1;
        }
        
        for(int i = 0;i<len - 1;i++){
            if(ratings[i] < ratings[i+1]){
                dp[i+1] = Math.max(dp[i+1], dp[i] + 1);
            }
        }
        
        for(int i = len - 1;i>0;i--){
            if(ratings[i] < ratings[i-1]){
                dp[i-1] = Math.max(dp[i-1], dp[i] + 1);
            }
        }
        
        int result = 0;
        for(int i = 0;i<len;i++){
            result += dp[i];
        }
        return result;
    }
}