leetcode算法题--最大平均值和的分组★

原题链接：https://leetcode-cn.com/problems/largest-sum-of-averages/

动态规划

dp[i][k]表示A[:i]的k个组的和

状态转移：

dp[i][k]=min(dp[i][k],dp[j][k-1]+average(j,i)); 0<=j<i

代码：

double largestSumOfAverages(vector<int>& A, int K) {
    int len=A.size();
    vector<vector<double>> dp(len+1,vector<double>(K+1,DBL_MIN));
    vector<double> sum(len+1,0);
    for(int i=0;i<len;i++){
        sum[i+1]=sum[i]+A[i];//计算和
    }
    for(int i=1;i<=len;i++){
        dp[i][1]=sum[i]/i;//A[:i]和的平均数
    }
    for(int k=2;k<=K;k++){
        for(int i=1;i<=len;i++){
            for(int j=1;j<i;j++){
                dp[i][k]=max(dp[i][k],dp[j][k-1]+(sum[i]-sum[j])/(i-j));//状态转移
            }
        }
    }
    return dp[len][K];
}