最大平均值和的分组

/*
 * 思路:
 * 1. 动态规划的核心思想与普通Java解法相同。
 * 2. 主要区别在于使用Java Stream API来处理数组和计算前缀和。
 * 3. Stream API可以使代码更简洁，但对于性能提升不明显。
 */
import java.util.stream.IntStream;
public class Solution {
    public double largestSumOfAverages(int[] nums, int k) {
        int n = nums.length;
        double[] prefixSum = new double[n + 1];
        IntStream.range(0, n).forEach(i -> prefixSum[i + 1] = prefixSum[i] + nums[i]);
        double[][] dp = new double[n + 1][k + 1];
        IntStream.rangeClosed(1, n).forEach(i -> dp[i][1] = prefixSum[i] / i);
        IntStream.rangeClosed(2, k).forEach(j -> IntStream.rangeClosed(j, n).forEach(i -> IntStream.range(j - 1, i).forEach(m -> dp[i][j] = Math.max(dp[i][j], dp[m][j - 1] + (prefixSum[i] - prefixSum[m]) / (i - m)))));
        return dp[n][k];
    }
}