给你一个整数数组,返回它的某个 非空 子数组(连续元素)在执行一次可选的删除操作后,所能得到的最大元素总和。
换句话说,你可以从原数组中选出一个子数组,并可以决定要不要从中删除一个元素(只能删一次哦),(删除后)子数组中至少应当有一个元素,然后该子数组(剩下)的元素总和是所有子数组之中最大的。
注意,删除一个元素后,子数组 不能为空。
示例1
输入:
arr = [1,-2,0,3]
输出:
4
解释:
我们可以选出 [1, -2, 0, 3],然后删掉 -2,这样得到 [1, 0, 3],和最大。
示例2
输入:
arr = [1,-2,-2,3]
输出:
3
解释:
我们直接选出 [3],这就是最大和。
示例3
输入:
arr = [-1,-1,-1,-1]
输出:
-1
解释:
最后得到的子数组不能为空,所以我们不能选择 [-1] 并从中删去 -1 来得到 0。
我们应该直接选择 [-1],或者选择 [-1, -1] 再从中删去一个 -1。
提示
首先回顾一道很相似的题目,也就是求连续子数组的最大值,并不需要删除元素。
这其实只需要用动态规划就能实现了,也就是计算以 结尾的连续子数组的最大值,记为 。那么它一定要取 ,而前面的元素的话,如果 ,也就是以 为结尾的连续子数组最大值大于 0 ,那就加上前面的最大值,否则的话只取 就行了。最终答案就是取所有 中最大的,状态转移方程是:
回到本题,如果一个元素都不删除的话,那么做法就和上面一模一样。如果删除一个元素的话,那么它的左右两边就分成了两个连续的子数组了。
那么我们假设删除的是 ,那么我们只要求左右两边子数组的最大值之和,也就是以 结尾和以 开头的两个连续子数组的最大值之和。以 结尾上面已经求过了,以 开头和上面方法类似,从后往前求一遍就行了。这样预处理完两个动态规划数组之后,遍历删除的元素,就能 时间内算出最大值。
具体实现的时候,注意到删除的元素是有限制的,其实只需要遍历删除 到 就行了,因为删除首尾两个元素的话,剩下来一个子数组,答案已经包含在开始的预处理之中了。
空间方面,从右往左求以 开头的连续子数组最大值的时候,没有必要保存到数组里了,直接用变量保存,然后同时计算删除 之后最大值就行了。
class Solution {
public:
int maximumSum(vector<int>& arr) {
int n = arr.size();
int dp[n];
dp[0] = arr[0];
int res = dp[0];
for (int i = 1; i < n; ++i) {
dp[i] = arr[i] + max(dp[i-1], 0);
res = max(res, dp[i]);
}
int last = arr[n-1];
for (int i = n-2; i > 0; --i) {
res = max(res, dp[i-1]+last);
last = arr[i] + max(last, 0);
}
return res;
}
};
class Solution:
def maximumSum(self, arr: List[int]) -> int:
n = len(arr)
dp = [arr[0]] * n
for i in range(1, n):
dp[i] = arr[i] + max(dp[i-1], 0)
res, last = max(dp), arr[-1]
for i in range(n-2, 0, -1):
res = max(res, dp[i-1]+last)
last = arr[i] + max(last, 0)
return res
作者简介:godweiyang,知乎同名,华东师范大学计算机系硕士在读,方向自然语言处理与深度学习。喜欢与人分享技术与知识,期待与你的进一步交流~