给定数组 nums 和一个整数 k 。我们将给定的数组 nums 分成 最多 k 个相邻的非空子数组 。 分数 由每个子数组内的平均值的总和构成。
注意我们必须使用 nums 数组中的每一个数进行分组,并且分数不一定需要是整数。
返回我们所能得到的最大 分数 是多少。答案误差在 10-6 内被视为是正确的。
示例 1:
输入: nums = [9,1,2,3,9], k = 3
输出: 20.00000
解释:
nums 的最优分组是[9], [1, 2, 3], [9]. 得到的分数是 9 + (1 + 2 + 3) / 3 + 9 = 20.
我们也可以把 nums 分成[9, 1], [2], [3, 9].
这样的分组得到的分数为 5 + 2 + 6 = 13, 但不是最大值.
示例 2:
输入: nums = [1,2,3,4,5,6,7], k = 4
输出: 20.50000
提示:
1 <= nums.length <= 100
1 <= nums[i] <= 104
1 <= k <= nums.length
https://leetcode.cn/problems/largest-sum-of-averages/
nums = [9,1,2,3,9], k = 3
[9]、[1,2,3,9] = dp[0][2]+avg(1,2,3,9) = a (9分割成2份的答案+9后面的那份 = 整体分割成3份的答案)
[9,1]、[2,3,9] = dp[1][2]+avg(2,3,9) = b ([9,1]分割成2份的答案+[9,1]后面的那份 = 整体分割成3份的答案)
[9,1,2]、[3,9] = dp[2][2]+avg(3,9) = c ([9,1,2]分割成2份的答案+[9,1,2]后面的那份 = 整体分割成3份的答案)
[9,1,2,3]、[9] = dp[3][2]+avg(9) = d ([9,1,2,3]分割成2份的答案+[9,1,2,3]后面的那份 = 整体分割成3份的答案)所以dp[3][3] = max( a,b,c,d )
这里你可以认为dp[0][2]、dp[1][2]、dp[2][2]、dp[3][2]已经求出了。
有不懂的地方看注释就明白了
class Solution {
public:double largestSumOfAverages(vector& nums, int k) {//状态:数组A的每个元素,分割为K个相邻的数组//选择:枚举所有分割k的可能性,分割成k份 == 分割成k-1份+最后一份int n=nums.size();vector sum(nums.size()+1,0.0);//前缀和,要求平均值所以用doublevector> dp(nums.size()+1,vector(k+1,0));//第一维是前x个数,第二位是x个数分为k份的结果。for(int i =1;i<=n;++i) sum[i]=sum[i-1]+nums[i-1];//前缀和,求平均值用for(int i =1;i<=n;++i){dp[i][1]=sum[i]/i;for(int K=2;K<=k&&K<=i;++K)for(int j=1;j
时间复杂度:O(k∗n2k*n^2k∗n2)//三层循环
空间复杂度:O(k*n)//二维数组
class Solution {
public:double largestSumOfAverages(vector& nums, int k) {int n = nums.size();vector prefix(n + 1);for (int i = 0; i < n; i++) {prefix[i + 1] = prefix[i] + nums[i];}vector 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];}
};
时间复杂度:O(k∗n2k*n^2k∗n2)//三层循环
空间复杂度:O(n)//一维数组
上一篇:【Vue】Vuex-store
下一篇:四个字的成语偏旁部首都一样的