【Lintcode】620. Maximum Subarray IV

题目地址：

https://www.lintcode.com/problem/maximum-subarray-iv/description

给定一个长$n$的数组$A$，再给定一个正整数$k$，求其长度不小于$k$的和最大的子数组，返回这个最大和。如果$n<k$则返回$0$。

思路是前缀和。如果$n<k$则直接返回$0$。否则先求一下前缀和数组$p$，即$p[i]=\sum A[0:i-1]$。接着从$p[k]$开始向后遍历，每次遍历到$p[i]$的时候，同时用一个变量维护$p[0:i-k]$的最小值$m$，这样遍历到$p[i]$的时候，就能得到以$A[i-1]$为右端点的长不小于$k$的子数组的最大和，那就是$p[i]-m$。用该数更新答案即可。代码如下：

public class Solution {
    /**
     * @param nums: an array of integer
     * @param k: an integer
     * @return: the largest sum
     */
    public int maxSubarray4(int[] nums, int k) {
        // write your code here
        // 特判一下
        if (nums.length < k) {
            return 0;
        }
        
        // 求一下前缀和数组
        int[] preSum = new int[nums.length + 1];
        for (int i = 0; i < nums.length; i++) {
            preSum[i + 1] = preSum[i] + nums[i];
        }
        
        // curMin存preSum[0 : i - k]的最小值
        int res = Integer.MIN_VALUE, curMin = Integer.MAX_VALUE;
        for (int i = k; i <= nums.length; i++) {
            curMin = Math.min(curMin, preSum[i - k]);
            res = Math.max(res, preSum[i] - curMin);
        }
        
        return res;
    }
}

时空复杂度$O(n)$。