Ai
1 Star 2 Fork 5

LilithSangreal/LeetCode-Solutions

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
克隆/下载
maximum-sum-of-3-non-overlapping-subarrays.cpp 1.46 KB
一键复制 编辑 原始数据 按行查看 历史
kamyu 提交于 2017-10-01 17:37 +08:00 . Create maximum-sum-of-3-non-overlapping-subarrays.cpp
// 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;
}
};
Loading...
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/LilithSangreal/LeetCode-Solutions.git
git@gitee.com:LilithSangreal/LeetCode-Solutions.git
LilithSangreal
LeetCode-Solutions
LeetCode-Solutions
master

搜索帮助