《剑指offer》面试题42. 连续子数组的最大和

问题描述

输入一个整型数组，数组里有正数也有负数。数组中的一个或连续多个整数组成一个子数组。求所有子数组的和的最大值。
要求时间复杂度为O(n)。
示例1:
输入: nums = [-2,1,-3,4,-1,2,1,-5,4]
输出: 6
解释: 连续子数组 [4,-1,2,1] 的和最大，为 6。
 
提示：
1 <= arr.length <= 10^5
-100 <= arr[i] <= 100

代码

我们定义\(dp[i]\)是以元素\(nums[i]\)结尾的最大和，则更新公式\(dp[i] = max(nums[i],nums[i]+dp[i-1])\)

class Solution {
public:
    int maxSubArray(vector<int>& nums) {
        int i, n = nums.size(),ans = nums[0];
        vector<int> dp(n);
        dp[0] = nums[0];
        for(i = 1; i < n; ++i)
        {
            dp[i] = max(nums[i],nums[i]+dp[i-1]);
            ans = max(dp[i],ans);
        }
        return ans;
    }
};

结果

执行用时 :52 ms, 在所有 C++ 提交中击败了25.93%的用户
内存消耗 :23.8 MB, 在所有 C++ 提交中击败了100.00%的用户

代码

当然我们也可以把空间复杂度降为\(O(1)\),根据上面的更新公式，显然：

\[dp[i] = \begin{cases}
dp[i-1]+nums[i],&dp[i-1]>0\\
nums[i],&dp[i-1]\leq 0
\end{cases}
\]

因此我们可以直接在\(nums\)上直接修改

class Solution {
public:
    int maxSubArray(vector<int>& nums) {
        int i, n = nums.size(),ans = nums[0];
        for(i = 1; i < n; ++i)
        {
            nums[i] += max(0,nums[i-1]);
            ans = max(nums[i],ans);
        }
        return ans;
    }
};

结果：

执行用时 :44 ms, 在所有 C++ 提交中击败了42.46%的用户
内存消耗 :22.9 MB, 在所有 C++ 提交中击败了100.00%的用户

• 返回目录