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

题目描述:

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

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

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

Level AC rate
Medium 58.7%

题目解析:

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

class Solution {
public:
    double largestSumOfAverages(vector& nums, int k) {
        int len = nums.size();
        vector pre(len+1);
        for(int i = 0 ; i < len ; i++){
            pre[i+1] = pre[i]+nums[i];
        }
        vector> dp(len+1,vector(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%的用户

你可能感兴趣的:(每日一题,leetcode,算法,数据结构,c++)