题目描述

输入一个整型数组,数组里有正数也有负数。数组中的一个或连续多个组成一个子数组。求所有子数组的和的最大值。要求时间复杂度为O(n)。

例如输入的数组为{1, -2, 3, 10, -4, 7, 2, -5},和最大的子数组为{3, 10, -4, 7, 2}。因此输出为该子数组的和18 。

解题思路

举例分析数组的规律

我们试着从头到尾逐个累加示例数组中的每个数字。初始化和为0。第一步加上第一个数字1, 此时和为1。接下来第二步加上数字-2,和就变成了-1。第三步刷上数字3。我们注意到由于此前累计的和是-1 ,小于0,那如果用-1 加上3 ,得到的和是2 , 比3 本身还小。也就是说从第一个数字开始的子数组的和会小于从第三个数字开始的子数组的和。因此我们不用考虑从第一个数字开始的子数组,之前累计的和也被抛弃。

我们从第三个数字重新开始累加,此时得到的和是3 。接下来第四步加10,得到和为13 。第五步加上-4, 和为9。我们发现由于-4 是一个负数,因此累加-4 之后得到的和比原来的和还要小。因此我们要把之前得到的和13 保存下来,它有可能是最大的子数组的和。第六步加上数字.7,9 加7 的结果是16,此时和比之前最大的和13 还要大, 把最大的子数组的和由13更新为16。第七步加上2,累加得到的和为18,同时我们也要更新最大子数组的和。第八步加上最后一个数字-5,由于得到的和为13 ,小于此前最大的和18,因此最终最大的子数组的和为18 ,对应的子数组是{3, 10, -4, 7, 2}。

应用动态规划法

我们用函数f(i)来表示以第i个数字结尾的子数组的最大和,那么我们需要求出max[f(i)], 其中0<=i<n,我们可以用如下递归公式求f(i):

f(i) = array[i]              i = 0 或者 f(i) <= 0
f(i) = f(i-1) + array[i]     i > 0 并且 f(i) > 0

如果我们用一个数组来表示f(i),那么最后我们又要排序,所以又在无形之间加大了时间复杂度,所以在进行上述过程中就需要加入比较逻辑,当上述过程完成之后,最大值也得到了。

代码

public class SubArrayProcess {
    public static int findGreatestSumOfSubArray(int[] data) {
        // 参数校验
        if (data == null || data.length == 0) {
            return 0;
        }

        // 当前的和
        int curSum = 0;
        // 记录最大的子数组和,开始时是最小的整数
        int greatestSum = Integer.MIN_VALUE;

        for (int i = 0; i < data.length; i++) {
            // 如果当前和小于等于0,就重新设置当前和
            if (curSum <= 0) {
                curSum = data[i];
            }
            // 如果当前和大于0,累加当前和
            else {
                curSum += data[i];
            }

            // 更新记录到的最大子数组和
            if (curSum > greatestSum) {
                greatestSum = curSum;
            }
        }

        return greatestSum;
    }

    // 测试
    public static void main(String[] args) {
        int[] data = {1,-2,3,10,-4,7,2,-5};
        System.out.println(findGreatestSumOfSubArray(data));
        
    }
}