【Lintcode】1619. Candy II

题目地址：

https://www.lintcode.com/problem/candy-ii/description

给定一个数组$A$，代表每个小朋友的评分。现在要给每个小朋友分糖果，规则是：
1、每个小朋友至少要分得$1$个糖果；
2、并且如果某个小朋友的评分比他相邻的小朋友评分高，那么他就应该比他相邻的那个小朋友得到的糖果更多；
3、如果某个小朋友评分和他相邻的小朋友相等，那么他们应该得到相同数量的糖果。
问总共至少需要多少个糖果。

思路是动态规划。某个下标$i$的小朋友得到的糖果数应该是数组$A$中从$i$出发向两边非严格下降的路径长度中长度更长的那个（相等的数视为只有一个）。例如对于数组$A=[1,6,6,2]$，那么下标$2$的小朋友向左的下降路径长度是$(1,6)$的长度$2$（相同数视为只出现了一次），向右的下降路径长度是$2$，所以他的糖果数就是$2$。算法正确性证明不难（可以参考https://blog.csdn.net/qq_46105170/article/details/108633071），此处略去。接下来考虑怎么算。设$f[i]$是以$A[i]$为结尾的从左边的最长上升路径的长度，那么$$f[i]=\{\begin{array}{l}1,i=0\\ f[i-1],A[i]=A[i-1]\\ 1+f[i-1],A[i]>A[i-1]\end{array}$$类似可以设$g[i]$为以$A[i]$为结尾的从右边的最长上升路径的长度，递推方程类似。则下标为$i$的小朋友的糖果数就是 max ⁡ { f [ i ] , g [ i ] } \max\{f[i],g[i]\} max{f[i],g[i]}。累加求和即可。代码如下：

import java.util.Arrays;

public class Solution {
    /**
     * @param ratings: rating value of each child
     * @return: Return the minimum candies you must give.
     */
    public int candyII(int[] ratings) {
        // Write your code here
        int len = ratings.length;
        int[] dpLeft = new int[len], dpRight = new int[len];
    
        Arrays.fill(dpLeft, 1);
        Arrays.fill(dpRight, 1);
    
        for (int i = 1; i < ratings.length; i++) {
            if (ratings[i] == ratings[i - 1]) {
                dpLeft[i] = dpLeft[i - 1];
            } else if (ratings[i] > ratings[i - 1]) {
                dpLeft[i] = 1 + dpLeft[i - 1];
            }
        }
    
        for (int i = ratings.length - 2; i >= 0; i--) {
            if (ratings[i] == ratings[i + 1]) {
                dpRight[i] = dpRight[i + 1];
            } else if (ratings[i] > ratings[i + 1]) {
                dpRight[i] = 1 + dpRight[i + 1];
            }
        }
        
        int res = 0;
        for (int i = 0; i < len; i++) {
            res += Math.max(dpLeft[i], dpRight[i]);
        }
        
        return res;
    }
}

时空复杂度$O(n)$。