leetcode hot100 之 最大子数组和

题目

给定一个数组，找出最大连续子数组的和。

输入：nums = [-2,1,-3,4,-1,2,1,-5,4]
输出：6
解释：连续子数组 [4,-1,2,1] 的和最大，为 6 。

原题链接： https://leetcode-cn.com/problems/maximum-subarray/

思路

设定 dp[i] 表示，以当前节点为终点，最大的连续子数组的和。
转移方程为 dp[i] = max(dp[i-1] + nums[i], nums[i])。
也就是说，如果前面的子数组是正的，那么就把前面的子数组和当前节点拼起来。否则子数组又从当前节点重新开始。
初始化 dp[0] = nums[0]。
由于转移方程为1维，且dp[i] 只依赖 dp[i-1]，所以这里其实只需要维护 dp[1]、dp[0] 即可。

• 复杂度分析
  • 时间复杂度 O(n)
  • 空间复杂度 O(1)

代码

class Solution {
public:
    int maxSubArray(vector<int>& nums) {
        if (nums.empty()) {
            return 0;
        }
        int dp0 = nums[0];
        int dp1;
        int max_sum = dp0;
        for (int i = 1; i < nums.size(); i++) {
            dp1 = max(dp0 + nums[i], nums[i]);
            max_sum = max(max_sum, dp1);
            dp0 = dp1;
        }
        return max_sum;
    }
};