lintcode(412)分糖果

来源:互联网 发布:海沧广电网络客服 编辑:程序博客网 时间:2024/06/05 06:22

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;    }}