135. 分发糖果

package com.heu.wsq.leetcode;

/**
 * 135. 分发糖果
 * @author wsq
 * @date 2020/12/24
 * 老师想给孩子们分发糖果，有 N 个孩子站成了一条直线，老师会根据每个孩子的表现，预先给他们评分。
 * 你需要按照以下要求，帮助老师给这些孩子分发糖果：
 * 每个孩子至少分配到 1 个糖果。
 * 相邻的孩子中，评分高的孩子必须获得更多的糖果。
 * 那么这样下来，老师至少需要准备多少颗糖果呢？
 *
 * 示例 1:
 * 输入: [1,0,2]
 * 输出: 5
 * 解释: 你可以分别给这三个孩子分发 2、1、2 颗糖果。
 *
 * 链接：https://leetcode-cn.com/problems/candy
 */
public class Candy {
    /**
     * n*n的时间复杂度，在每一个位置都需要从左右两方向寻找比当前位置分数低的同学，最后取左右两边糖果数最大的一个加上当前位置的糖果数
     * @param ratings
     * @return
     */
    public int candy(int[] ratings){
        int n = ratings.length;
        if (n == 0 || n == 1){
            return n;
        }
        int[] f = new int[n];
        for (int i = 0; i < n; i++){
            f[i] = 1;
            int last = ratings[i];
            int tmp1 = 0;
            for (int j = i-1; j >= 0; j--){
                if (ratings[j] >= last){
                    break;
                }
                last = ratings[j];
                tmp1++;
            }
            last = ratings[i];
            int tmp2 = 0;
            for (int j = i + 1; j < n; j++){
                if (ratings[j] >= last){
                    break;
                }
                last = ratings[j];
                tmp2++;
            }
            f[i] += Math.max(tmp1, tmp2);
        }
        int ans = 0;
        for (int i = 0; i < n; i++){
            ans += f[i];
        }
        return ans;
    }

    public static void main(String[] args) {
        int[] ratings = {1, 2, 2};
        Candy candy = new Candy();
        int ans = candy.candy(ratings);
        System.out.println(ans);
    }
}