leetcode813. 最大平均值和的分组，动态规划

leetcode813. 最大平均值和的分组,动态规划题。

题目

给定数组 nums 和一个整数 k 。我们将给定的数组 nums 分成 最多 k 个相邻的非空子数组 。 分数 由每个子数组内的平均值的总和构成。
注意我们必须使用 nums 数组中的每一个数进行分组，并且分数不一定需要是整数。
返回我们所能得到的最大 分数 是多少。答案误差在 10-6 内被视为是正确的。

解题思路

动态规划求解，定义数组dp[i][j]表示数组中前i个字符分成j个部分时的各个部分的最大平均值和。状态转移方程如下图：

即当j等于1时，dp[i][j]的值就等于前i个数字的平均值。当j>1时，dp[i][j]等于将前i个数字分为[0,x]和[x,i]两个部分，也就等于dp[x][j-1]加上
[x,i]这部分的平均值。然后遍历所有状态返回dp[n][k]即可得到最终值。
由于状态转移过程中只涉及到j-1的转移，所以可以用滚动数组只用dp[i]来记录当前j的值情况下dp[i][j]的值减少空间复杂度，即将i从最大值n开始处理，这样就避免了更新状态时会影响后面dp值。

代码

class Solution {
public:
    double largestSumOfAverages(vector<int>& nums, int k) {
        int n=nums.size();
        vector<double> dp(n);
        vector<double> pre(n);//前缀和，方便计算平均值
        pre[0]=nums[0];
        for(int i=0;i<n;i++)
        {
            if(i)pre[i]=pre[i-1]+nums[i];
            dp[i]=pre[i]/(i+1);
        }
        for(int j=2;j<=k;j++)
        {
            for(int i=n-1;i>=j-1;i--)//分成j份，至少需要j个数字，所以i>=j-1
            {
                for(int x=j-2;x<i;x++)
                {
                    dp[i]=max(dp[i], dp[x]+(pre[i]-pre[x])/(i-x));
                }
            }
        }
        return dp[n-1];
    }
};