6/19/2014

134. Candy

public class Solution {
    public int candy(int[] ratings) {
        if (ratings.length==0 || ratings==null) return 0;
       
        int[] count = new int[ratings.length];
        Arrays.fill(count, 1);
        int sum = 0;
       
        for (int i=1; i<count.length; i++) {
            if (ratings[i]>ratings[i-1]) {
                count[i] = count[i-1] +1;
            }
        }
       
        for (int i=count.length-1; i>=1; i--) {
            sum += count[i];
            if (ratings[i-1]>ratings[i] && count[i-1]<=count[i]) {
                count[i-1] = count[i]+1;
            }
        }
       
        sum += count[0];
       
        return sum;
    }
}

没有评论:

发表评论