[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 位有符号整数范围内。

题目分类

数学 数组 前缀和

思路

第一种

第j个前缀和减去第i个前缀和加上第i个数就可以遍历数组的所有长度大于等于2的连续子数组和。

第二种

只要存在两个前缀和除以k的余数相同，那两者前缀和的差一定是k的倍数。使用HashMap来优化算法。

注意事项

k为0时的情况。

代码

第一种

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) {
        int len = nums.size();
        if(len <= 1) return false;
        vector<int> dp(len, 0);
        dp[0] = nums[0];
        for(int i = 1; i < len; ++i) {
            dp[i] = dp[i-1] + nums[i];
        }
        for(int i = 0; i < len; ++i) {
            for(int j = i+1; j < len; ++j) {
                if(k == 0) { 
                    if (dp[j] - dp[i] + nums[i] == 0) {
                        return true;
                    }
                } else if((dp[j]-dp[i] + nums[i])  % k == 0) {
                    return true;
                }
            }
        }
        return false;
    }
};

第二种

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) {
        int sum = 0;
        unordered_map<int, int> m;
        m.insert({0, -1});
        for(int i = 0; i < nums.size(); ++i) {
            sum += nums[i];
            if(k != 0) sum = sum % k;
            if(m.count(sum)) {
                if(i - m[sum] > 1) {
                    return true;
                }
            }else {
                 m.insert({sum, i});
            } 
        }
        return false;
    }
};