leetcode.813 最大平均值和的分组 - 前缀和 + dp

最新推荐文章于 2022-11-28 12:53:54 发布

Roye_ack

最新推荐文章于 2022-11-28 12:53:54 发布

阅读量124

点赞数

分类专栏： leetcode每日一题文章标签：动态规划算法

本文链接：https://blog.csdn.net/weixin_61639349/article/details/128076781

版权

leetcode每日一题专栏收录该内容

79 篇文章 1 订阅

订阅专栏

813. 最大平均值和的分组

思路：

一个简单的数学结论：

划分份数越多，平均值之和越大，因此想要取得最大值必然是恰好划分成 k 份
用 $\large dp[i][j]$ 表示前i个元素划分成j份的最大平均和，则答案就是 $\large dp[n][k]$
当 $\large j=1$ 时， $\large dp[i][j]$ 就是 $\large [0,i-1]$ 元素的平均值
当 $\large j>1$ 时，我们可以把区间 $\large [0,i-1]$ 划分成 $\large [0,x-1]$ 和 $\large [x,i-1]$ 两部分， $\large dp[i][j]$ 等于所有这些合法的切分方式的平均值和的最大值
所以状态转移方程就是
dp[i][j]=max(dp[i][j],dp[x][j-1]+(s[i]-s[x])/(i-x));
其中 dp[x][j - 1] + (s[i] - s[x]) / (i - x) 表示为 x 位置之前的最大值分组和 + 当前位置平均值

class Solution {
public:
    double largestSumOfAverages(vector<int>& nums, int k) 
    {
        int n = nums.size();
        vector<double> s(n+1);
        for(int i=0;i<n;i++) s[i+1]=s[i]+nums[i];
        vector<vector<double>> dp(n+1,vector<double>(k+1));

        for(int i=1;i<=n;i++) dp[i][1]=s[i]/i;

        for(int j=2;j<=k;j++) //枚举分组数
            for(int i=j;i<=n;i++) //枚举分组元素
                for(int x=j-1;x<i;x++) //枚举分割位置
                    dp[i][j]=max(dp[i][j],dp[x][j-1]+(s[i]-s[x])/(i-x));
        return dp[n][k];
    }
};