LeetCode 523. 连续的子数组和（C++、python）

给定一个包含非负数的数组和一个目标整数 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 位有符号整数范围内。

C++

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) 
    {
        int n=nums.size();
        map<int,int> tmp;
        int val=0;
        tmp[0]=-1;
        for(int i=0;i<n;i++)
        {
            val+=nums[i];
            if(0==k)
            {
                if(0==nums[i])
                {
                    if(i>=1 && 0==nums[i-1])
                    {
                        return true;
                    }
                }
            }
            else
            {
                int mod=val%k;
                if(tmp.find(mod)!=tmp.end())
                {
                    int len=i-tmp[mod];
                    if(len>=2)
                    {
                        return true;
                    }
                }
                else
                {
                    tmp[mod]=i;
                }                
            }
        }
        return false;
    }
};

python

class Solution:
    def checkSubarraySum(self, nums: List[int], k: int) -> bool:
        n=len(nums)
        dic={0:-1}
        val=0
        for i in range(n):
            val+=nums[i]
            if 0==k:
                if 0==nums[i]:
                    if i>=1 and 0==nums[i-1]:
                        return True
            else:
                mod=val%k
                if mod not in dic:
                    dic[mod]=i
                else:
                    length=i-dic[mod]
                    if length>=2:
                        return True
        return False