代码拉取完成,页面将自动刷新
// Time: O(n)
// Space: O(n)
class Solution {
public:
vector<int> maxSumOfThreeSubarrays(vector<int>& nums, int k) {
const auto n = nums.size();
vector<int> accu = {0};
for (const auto& num : nums) {
accu.emplace_back(accu.back() + num);
}
vector<int> left_pos(n);
for (int i = k, total = accu[k] - accu[0]; i < n; ++i) {
if (accu[i + 1] - accu[i + 1 - k] > total) {
left_pos[i] = i + 1 - k;
total = accu[i + 1] - accu[i + 1 - k];
} else {
left_pos[i] = left_pos[i - 1];
}
}
vector<int> right_pos(n, n - k);
for (int i = n - k - 1, total = accu[n] - accu[n - k]; i >= 0; --i) {
if (accu[i + k] - accu[i] > total) {
right_pos[i] = i;
total = accu[i + k] - accu[i];
} else {
right_pos[i] = right_pos[i + 1];
}
}
vector<int> result(3);
for (int i = k, max_sum = 0; i <= n - 2 * k; ++i) {
auto left = left_pos[i - 1], right = right_pos[i + k];
auto total = (accu[i + k] - accu[i]) +
(accu[left + k] - accu[left]) +
(accu[right + k] - accu[right]);
if (total > max_sum) {
max_sum = total;
result = {left, i, right};
}
}
return result;
}
};
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。