leetcode 1186.删除一次得到子数组最大和

题目描述

原题链接：1186.删除一次得子数组最大和

解题思路

我们先考虑这样一个问题：如果将题意中的“删除一次”的约束删去，那么本题又该如何做呢？

1.暴力做法($o(n^2)$)

首先，我们的直觉肯定是先用暴力，枚举每一个点$i$，在固定$i$的前提下，枚举每一个$j$，算出$[i,j]$之间的子数组的和。这样枚举完所有的$i$和$j$，我们就可以得到子数组最大和。但是，这样做的时间复杂度是$o(n^2)$，很显然会超出时间限制。那么有没有更好的方法呢？

2.Kadane算法($(o(n)$)

Kadane算法利用了动态规划的算法，只用遍历一遍数组即可找到子数组的最大和。
具体来说，我们定义一个数组$dp[i]$表示以第$i$个元素结尾的子数组的最大和。这样我们就可以分两种情况：

1. 保留前面$i-1$个元素得到的子数组
2. 扔掉前面的的$i-1$个元素，以第$i$个元素作为起始元素。

这样，转移方程就可以表示为：
$$dp[i]=max(dp[i-1]+arr[i],arr[i])$$
我们再利用一个global变量存储最大的dp[i]就可以实现一遍遍历得到答案啦！

Kadane算法代码

int Kadane(vector<int>& arr) {
    int cur = arr[0], glo = arr[0];
    for(int i = 1; i < arr.size(); i++) {
        cur = max(cur+arr[i], arr[i]);
        glo = max(glo, cur);
    }
    return glo;
}

为了节省存储资源，我们将数组dp[i]换成变量cur。如果这样做，一定要注意从小到大遍历。

说完了Kadane算法，我们再来看看本题，本题可以删去子数组中的一个元素或不删去。根据上述条件，我们可以考虑建立$dp[i][0]$表示以第$i$个元素结尾的而且不删去元素的子数组的最大和，以$dp[i][1]$表示第$i$个元素结尾而且删去一个元素的子数组的最大和。
$dp[i][0]$的更新方式和上述Kadane算法相同，而$dp[i][1]$是由两种情况更新来的：

1. 选择删去前$i-1$个元素的子数组中的一个元素
2. 删去第$i$个元素

这样$dp[i][1]$的转移方程就可以表示为：
$$dp[i][1]=max(dp[i][1]+arr[i],dp[i][0])$$

C++代码

class Solution {
public:
    int maximumSum(vector<int>& arr) {
        int dp0 = arr[0], dp1 = 0, ans = arr[0];
        for(int i = 1; i < arr.size(); i++) {
            dp1 = max(dp1+arr[i], dp0);
            dp0 = max(dp0+arr[i], arr[i]);
            ans = max(ans, dp1);
            ans = max(ans, dp0);
        }
        return ans;
    }
};