代码练习之连续子数组的最大和

题目描述

给你一个整数数组 $nums$ ，请你找出一个具有最大和的连续子数组（子数组最少包含一个元素），返回其最大和。

子数组是数组中的一个连续部分。

例如输入 $nums=[-2,1,-3,4,-1,2,1,-5,4]$，输出应该是 $6$。因为连续子数组 $[4,-1,2,1]$ 的和最大，为 $6$。

思路

很容易想到可以用动态规划去解。

假设 $nums$ 数组的长度是 $n$，下标从 $0$ 到 $n-1$。

我们用 $f(i)$ 代表以 $nums[i]$ 结尾的连续子数组的最大和，那么最终的答案应该是
$$\underset{i=0}{\overset{n-1}{\max }}f(i)$$

求 $f(i)$ 时，我们可以考虑将 $nums[i]$ 单独作为一段子数组或是把它加入到 $f(i-1)$ 代表的那段以 $nums[i-1]$ 结尾的子数组中。由于求的是最大的连续子数组和，所以我们要取这两者的最大值。从而得出状态转移方程：
f ( i ) = max ⁡ { f ( i − 1 ) + n u m s [ i ] , n u m s [ i ] } f(i) = \max\{f(i - 1) + nums[i], nums[i]\} f(i)=max{f(i−1)+nums[i],nums[i]}

代码

public class Solution {
    public static void main(String[] args) {
        int[] nums = {-2, 1, -3, 4, -1, 2, 1, -5, 4};
        System.out.println(maxSum(nums));
    }

    /**
     * 求数组最长的连续子数组的和
     * @param nums 数组
     * @return 最长连续子数组和
     */
    private static int maxSum(int[] nums) {
        int n = nums.length;

        int[] dp = new int[n];
        dp[0] = nums[0];
        int maxSum = dp[0];
        for (int i = 1; i < n; i++) {
            dp[i] = Math.max(dp[i - 1] + nums[i], nums[i]);
            maxSum = Math.max(maxSum, dp[i]);
        }

        return maxSum;
    }
}

我们发现每次状态转移（$dp[i]$）的时候，只需要知道上一个状态（$dp[i-1]$）即可完成转移。所以不需要记录所有的状态。这样可以大大减少空间的消耗。修改后的代码如下：

public class Solution {
    public static void main(String[] args) {
        int[] nums = {-2, 1, -3, 4, -1, 2, 1, -5, 4};
        System.out.println(maxSum(nums));
    }

    /**
     * 求数组最长的连续子数组的和
     * @param nums 数组
     * @return 最长连续子数组和
     */
    private static int maxSum(int[] nums) {
   		// 用pre代替 dp 数组，只需记录上一个状态即可
        int pre = nums[0];
        int maxSum = pre;
        for (int num : nums) {
            pre = Math.max(pre + num, num);
            maxSum = Math.max(maxSum, pre);
        }

        return maxSum;
    }
}