Leetcode 813. Largest Sum of Averages

最新推荐文章于 2025-04-01 10:02:42 发布

转载最新推荐文章于 2025-04-01 10:02:42 发布 · 324 阅读

收录于

LeetCode

当前文章被收录于：

LeetCode

7 篇文章 0 人学习

订阅专栏查看详情

动态规划

当前文章被收录于：

动态规划

2 篇文章 0 人学习

订阅专栏查看详情

当前文章被以下社区和专栏收录：

本文介绍了一种使用动态规划解决最大平均值问题的方法。通过递归定义状态转移方程，实现对列表中整数分组以获得最大的平均值之和。此算法适用于将整数列表分割为K个连续子数组，以最大化各子数组平均值总和的场景。

转载自https://blog.csdn.net/birdreamer/article/details/79850052

class Solution:
    def largestSumOfAverages(self, A, K):
        """
        :type A: List[int]
        :type K: int
        :rtype: float
        """
        out = dict()

        def dp(length, groups):

            if (length, groups) in out: 
                return out[(length, groups)]

            if groups == 1:
                out[(length, groups)] = sum(A[:length]) / length
                return out[(length, groups)]

            temp = 0
            out[length, groups] = 0

            for i in range(length - 1, 0, -1):
                temp += A[i]
                out[(length, groups)] = max(out[(length, groups)], dp(i, groups - 1) + temp / (length - i))

            return out[(length, groups)]

        return dp(len(A), K)

利用动态规划，公式转自https://blog.csdn.net/qq_28954601/article/details/79852404