题解 | #分糖果问题#

分糖果时，定义一个数组存储每个小朋友的糖果数，分完求和即可

分数比两边都高，取两边最大的加1：  a[sta] = Math.max(candy(sta - 1, a, b, arr), candy(sta + 1, a, b, arr)) + 1;
分数比两边都低：取1即可
分数比左边高，右边低：左边加1  a[sta] = candy(sta - 1, a, b, arr) + 1;
分数比右边高，左边低：右边加1 a[sta] = candy(sta + 1, a, b, arr) + 1;
import java.util.*;
public class Solution {

/**
 * pick candy
 * @param arr int整型一维数组 the array
 * @return int整型
 */
 
public int candy (int[] arr) {
    int n = arr.length;
    int[] a = new int[n];
    boolean[] b = new boolean[n];
    int sum = 0;
    int min = n - 1;
    int c;
    for (int i = 0; i < n; i++) {
        candy(i, a, b, arr);
    }


    for (int i = 0; i < n; i++) {
        sum += a[i];
    }

    return sum;
    // write code here
}

public int candy(int sta, int[] a, boolean[] b, int[] arr) {
    if (b[sta]) {
        return a[sta];
    }

    int n = a.length - 1;
    if (sta == 0) {
        if (arr[0] <= arr[1]) {
            a[0] = 1;
        } else {
            a[0] = candy(1, a, b, arr) + 1;
        }
    } else if (sta == n) {
        if (arr[sta] <= arr[sta - 1]) {
            a[sta] = 1;
        } else {
            a[sta] = candy(sta - 1, a, b, arr) + 1;
        }
    } else if (arr[sta] <= arr[sta - 1] && arr[sta] <= arr[sta + 1]) {
        a[sta] = 1;
    } else if (arr[sta] > arr[sta - 1] && arr[sta] > arr[sta + 1]) {
        a[sta] = Math.max(candy(sta - 1, a, b, arr), candy(sta + 1, a, b, arr)) + 1;

    } else if (arr[sta] > arr[sta - 1] && arr[sta] <= arr[sta + 1]) {
        a[sta] = candy(sta - 1, a, b, arr) + 1;
    } else {
        a[sta] = candy(sta + 1, a, b, arr) + 1;
    }

    b[sta] = true;
    return a[sta];


}

}