给你一个整数数组,返回它的某个 非空 子数组(连续元素)在执行一次可选的删除操作后,所能得到的最大元素总和。
换句话说,你可以从原数组中选出一个子数组,并可以决定要不要从中删除一个元素(只能删一次哦),(删除后)子数组中至少应当有一个元素,然后该子数组(剩下)的元素总和是所有子数组之中最大的。
注意,删除一个元素后,子数组 不能为空。
请看示例:
示例 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。
提示:
1 <= arr.length <= 10^5
-10^4 <= arr[i] <= 10^4
解题思路
这个问题首先不难想到暴力解法,也就是将每个数字删除然后对于剩余的元素计算Leetcode 53:最大子序和(最详细的解法!!!),这么做显然会超时。
其实我们可以这么思考,我们只需要知道删除元素的左右最大区间和是多少,二者之和就是最后的结果。那么问题就变成了求当前数字结尾的最大区间和问题。实际上这依旧是Leetcode 53:最大子序和(最详细的解法!!!)。所以我们可以先正向计算一遍,例如[1,-2,0,3]
得到的当前数字结尾的最大区间和是:
1 -1 0 3
那么当前元素的左边最大区间和就是[1,-1,0]
。接着再反向计算一遍,也就是对于[3,0,-2,1]
,得到
3 3 1 2 → 2 1 3 3
那么当前元素的右边最大区间和就是[1,3,3]
。将二者求和的结果就是备选答案之一。
1 -1 0
1 3 3
+ ----------
4 -2
另外还存在一种情况,就是当前元素两边最大区间和的区间长度是0
,那么我们此时只有取当前元素。所以所有区间元素的最大值也是备选答案之一。另外还有两种情况是左边区间长度为0
和右边长度为0
,对于这两种情况我们分别求前面两个数组的最大值即可。那么最后的结果就是四者的最大值。所以答案来自以下四种情况:
- 只包含当前元素的区间
- 不包含当前元素的左区间
- 不包含当前元素的右区间
- 不包含当前元素的左右区间
代码如下
class Solution:
def maximumSum(self, arr: List[int]) -> int:
pre, pos = arr[::], arr[::-1]
cur_max = arr[0]
for i in range(1, len(pre)):
if pre[i-1] > 0:
pre[i] += pre[i-1]
if pos[i-1] > 0:
pos[i] += pos[i-1]
cur_max = max(cur_max, pre[i], pos[i])
return max([pre[i] + pos[-i-3] for i in range(len(pre)-2)] + [cur_max])
这个题使用动态规划的话就非常容易。总共只有两种状态,deleted
表示当前元素之前有元素删除了,notDeleted
表示当前位置之前没有位置删除,那么
- d e l e t e d = m a x ( d e l e t e d + a , a , n o t D e l e t e d ) deleted=max(deleted+a,a,notDeleted) deleted=max(deleted+a,a,notDeleted)
- n o t D e l e t e d = m a x ( n o t D e l e t e d + a , a ) notDeleted=max(notDeleted+a,a) notDeleted=max(notDeleted+a,a)
deleted
状态包含三个方面:以当前元素结尾并且之前是deleted
,以当前元素作为起始,不包含当前元素。而notDeleted
状态包含两个方面:以当前元素结尾并且之前状态是notDeleted
,以当前元素作为起始。最后的结果就是deleted
状态的最大值。
class Solution:
def maximumSum(self, arr: List[int]) -> int:
deleted = notDeleted = 0
res = float('-inf')
for i, a in enumerate(arr):
deleted = max(deleted + a, a)
if i > 0:
deleted = max(deleted, notDeleted)
notDeleted = max(notDeleted + a, a)
res = max(res, deleted)
return res
我将该问题的其他语言版本添加到了我的GitHub Leetcode
如有问题,希望大家指出!!!