[LeetCode] 523. 连续的子数组和 ☆☆☆(动态规划)

连续的子数组和 更高级解法

描述

给定一个包含非负数的数组和一个目标整数 k，编写一个函数来判断该数组是否含有连续的子数组，其大小至少为 2，总和为 k 的倍数，即总和为 n*k，其中 n 也是一个整数。

示例 1:

输入: [23,2,4,6,7], k = 6
输出: True
解释: [2,4] 是一个大小为 2 的子数组，并且和为 6。
示例 2:

输入: [23,2,6,4,7], k = 6
输出: True
解释: [23,2,6,4,7]是大小为 5 的子数组，并且和为 42。
说明:

数组的长度不会超过10,000。
你可以认为所有数字总和在 32 位有符号整数范围内。

解析

根据示例数组。

23 

25 2

29 6 4

35 12 10 6    可以看出规律。

使用一维数组，存储当前的值，看是否是k的整数倍。

代码

public static boolean checkSubarraySum(int[] nums, int k) {
        if (null == nums || nums.length <= 1) {//大小至少为2
            return false;
        }
        int[] dp = new int[nums.length + 1];
        for (int i = 0; i < nums.length; i++) {
            for (int kk = i; kk >= 0; kk--) {
                dp[kk] = dp[kk + 1] + nums[kk];
                if (kk < i && (dp[kk] == k || (k != 0 && dp[kk] % k == 0))) {
                    return true;
                }
            }
        }
        return false;
    }