题解 | #分糖果问题#TOP95

import java.util.*;


public class Solution {
    /**
     * pick candy
     * @param arr int整型一维数组 the array
     * @return int整型
     */
    public int candy (int[] arr) {
        // write code here
        if(arr == null || arr.length == 0){
            return 0;
        }
        int len = arr.length;
        int[] temp = new int[len];
        //初始化
        for(int i = 0;i < len;i++){
            temp[i] = 1;
        }
        //从左到右 如果大于则递增+1，多拿一颗糖
        for(int i = 1;i< len;i++){
            if(arr[i] > arr[i-1]){
                temp[i] = temp[i-1] + 1;
            }
        }
        int result = temp[len - 1];
        for(int i = len - 2 ;i>=0 ;i--){
            //如果左边的数大，并且左边的糖果小
            if(arr[i] > arr[i+1] && temp[i] <= temp[i+1]){
                temp[i] = temp[i+1]+1;
            }
            result += temp[i];
        }
        return result;
        
    }
}