LeetCode-813-最大平均值和的分组

最新推荐文章于 2024-10-27 15:16:59 发布

犬兄的海角

最新推荐文章于 2024-10-27 15:16:59 发布

阅读量299

点赞数

分类专栏： LeetCode刷题记录 LeetCode-待复习文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/weixin_43825194/article/details/128077703

版权

LeetCode刷题记录同时被 2 个专栏收录

312 篇文章 1 订阅

订阅专栏

LeetCode-待复习

145 篇文章 1 订阅

订阅专栏

在这里插入图片描述

1、动态规划法

我们可以利用数组 $d p [i] [j]$ 来表示我们将数组中区间 $[0, i - 1]$ 的元素分为 $j$ 组的平均值的总和。因此我们可以得到状态转化方程如下： $\left\{\begin{matrix} dp[i][j]= \frac{\textstyle \sum_{r=0}^{i-1}nums[r]}{i} ,j==1 \\ dp[i][j]= \underset{x\ge j-1}{max} (dp[x][j-1]+\frac{\textstyle \sum_{r=x}^{i-1}nums[r]}{i-x}),j>1 \end{matrix}\right.$
考虑到我们在状态转化方程中会经常使用到区间之和，因此我们可以使用前缀和数组来代替常用的数组。

class Solution {
public:
    double largestSumOfAverages(vector<int>& nums, int k) {
        int n = nums.size();
        vector<double> prefix(n + 1);
        for (int i = 0; i < n; i++) {
            prefix[i + 1] = prefix[i] + nums[i];
        }
        vector<vector<double>> dp(n + 1, vector<double>(k + 1));
        for (int i = 1; i <= n; i++) {
            dp[i][1] = prefix[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] + (prefix[i] - prefix[x]) / (i - x));
                }
            }
        }
        return dp[n][k];
    }
};

2、动态规划法优化

由于我们在数组中每次只使用到上一轮的数组，因此我们可以只是用一个一维数组来代替二维数组。

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