Leetcode 813. 最大平均值和的分组

题目描述:

给定数组 nums 和一个整数 k 。我们将给定的数组 nums 分成 最多 k 个相邻的非空子数组 。 分数 由每个子数组内的平均值的总和构成。

注意我们必须使用 nums 数组中的每一个数进行分组,并且分数不一定需要是整数。

返回我们所能得到的最大 分数 是多少。答案误差在 10-6 内被视为是正确的。

LevelAC rate
Medium58.7%

题目解析:

一个元素要么自成一组要么并入前面的组,怎么并入形成的平均值最大呢,如果该元素序号为i,那么就找前i个元素分为j-1组(不一定要完全包括i个元素)与剩余元素和当前元素之和的均值最大值即可,前缀和的动态规划,这里有一个值得注意的点是,for循环中的顺序,外层应该为组数j,内层为当前元素序号i,再内层为分为j-1组的元素数选择,因为每次计算需要知道j-1组的所有i值,代码如下:

class Solution {
public:
    double largestSumOfAverages(vector<int>& nums, int k) {
        int len = nums.size();
        vector<double> pre(len+1);
        for(int i = 0 ; i < len ; i++){
            pre[i+1] = pre[i]+nums[i];
        }
        vector<vector<double>> dp(len+1,vector<double>(k+1));
        for(int i = 0 ; i < len ; i++){
            dp[i+1][1] = pre[i+1]/(i+1);
        }
        for(int j = 2 ; j <= k ; j++){
            for(int i = j ; i <= len ; i++){
                for(int x = j-1 ; x < i ; x++){
                    dp[i][j] = max(dp[i][j],dp[x][j-1]+(pre[i]-pre[x])/(i-x));
                }
            }
        }
        return dp[len][k];
    }
};

执行用时:16 ms, 在所有 C++ 提交中击败了77.25%的用户

内存消耗:7.7 MB, 在所有 C++ 提交中击败了58.98%的用户

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值