假设我们有一个称为正整数的数组,我们必须找到三个具有最大和的非重叠子数组。在这里,每个子数组的大小为k,我们希望最大化所有3 * k项的总和。
我们必须以代表每个间隔的开始位置的索引列表的形式找到结果。如果有多个答案,我们将返回字典上最小的答案。
因此,如果输入像[1,2,1,2,6,8,4,1]且k = 2,则结果将为[0,3,5],因此子数组为[1,2], [2,6],[8,4]对应于起始索引[0,3,5]。
为了解决这个问题,我们将遵循以下步骤-
n:= nums的大小
定义大小为3的数组ret用inf填充
定义大小为n + 1的数组和
对于初始化i:= 0,当i <n时,更新(将i增加1),执行-
sum [i + 1] = sum [i] + nums [i]
定义大小为n的数组posLeft
定义大小为n的posRight数组,用n-k填充
对于初始化i:= k,currMax:= sum [k]-sum [0],当i <n时,更新(将i增加1),执行-
posLeft [i]:= posLeft [i-1]
currMax:= newTotal
posLeft [i]:= i +1-k
newTotal:= sum [i +1]-sum [i +1-k]
如果newTotal> currMax,则-
除此以外
对于初始化i:= n-k-1,currMax:= sum [n]-sum [n-k],当i> = 0时,更新(将i减1),-
posRight [i]:= posRight [i +1]
currMax:= newTotal
posRight [i]:= i
newTotal:= sum [i + k]-sum [i]
如果newTotal> = currMax,则-
除此以外
要求:= 0
对于初始化i:= k,当i <= n-2 * k时,更新(将i增加1),执行-
ret [0]:= l,ret [1]:= i,ret [2]:= r
req:=临时
l:= posLeft [i-1],r:= posRight [i + k]
温度:=(sum [l + k]-sum [l])+(sum [i + k]-sum [i])+(sum [r + k]-sum [r])
如果temp> req,则-
返回ret
让我们看下面的实现以更好地理解-
#include <bits/stdc++.h> using namespace std; void print_vector(vector<auto> v){ cout << "["; for(int i = 0; i<v.size(); i++){ cout << v[i] << ", "; } cout << "]"<<endl; } class Solution { public: vector<int> maxSumOfThreeSubarrays(vector<int>& nums, int k) { int n = nums.size(); vector <int> ret(3, INT_MAX); vector <int> sum(n + 1); for(int i = 0; i < n; i++){ sum[i + 1] = sum[i] + nums[i]; } vector <int> posLeft(n); vector <int> posRight(n, n - k); for(int i = k, currMax = sum[k] - sum[0]; i < n; i++){ int newTotal = sum[i + 1] - sum[i + 1- k]; if(newTotal > currMax){ currMax = newTotal; posLeft[i] = i + 1 - k; }else{ posLeft[i] = posLeft[i - 1]; } } for(int i = n - k - 1, currMax = sum[n] - sum[n - k]; i >=0 ; i--){ int newTotal = sum[i + k] - sum[i]; if(newTotal >= currMax){ currMax = newTotal; posRight[i] = i; }else{ posRight[i] = posRight[i + 1]; } } int req = 0; for(int i = k; i <= n - 2 * k; i++){ int l = posLeft[i - 1]; int r = posRight[i + k]; int temp = (sum[l + k] - sum[l]) + (sum[i + k] - sum[i]) + (sum[r + k] - sum[r]); if(temp > req){ ret[0] = l; ret[1] = i; ret[2] = r; req = temp; } } return ret; } }; main(){ Solution ob; vector<int> v = {1,2,1,2,6,8,4,1}; print_vector(ob.maxSumOfThreeSubarrays(v, 2)); }
{1,2,1,2,6,8,4,1} 2
输出结果
[0, 3, 5, ]