题解 | #分糖果问题#

题目 分糖果

日期 2022/03/19

思路

从左往后、从右往左各遍历一次

代码

public class Solution {
    /**
     * pick candy
     * @param arr int整型一维数组 the array
     * @return int整型
     */
    public int candy (int[] arr) {
        // 从左往后、从右往左各遍历一次
        int[] candys = new int[arr.length];
        int sum = 0;
        for(int i=0;i<candys.length;i++)
            candys[i] = 1;
        for(int i=0;i<arr.length-1;i++)
            if(arr[i] < arr[i+1])
                candys[i+1] = candys[i]+1;
        for(int i=arr.length-1;i>0;i--)
            if(arr[i-1]>arr[i] && candys[i-1]<=candys[i])
                candys[i-1] = candys[i]+1;
        for(int i:candys)
            sum+=i;
        return sum;
    }
}