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

---

思路

使用哈希map来存储累积和除以k后的余数。
为什么存余数？
遍历数组，将当前的累积和存储到map中，如果存在i-j范围内的连续子数组和是k的倍数，那么当前余数sum[j] % k和i-1时刻的余数sum[i-1] % k是一致的。sum[j] = sum[i-1] + n*k，所以map里存余数，如果发现重复余数，则存在连续子数组和是k的倍数。
注意题目中要求子数组长度大于1，map中key是余数，value就存储索引信息，遇到重复余数时，检查一下索引差值是否大于1（对应子数组长度大于1），大于则返回true。

注意一种特殊情况：数组总和刚好能整除k，在索引n-1时在map中首次添加key值：0，当前算法对于这种情况是返回false的，为解决这种冲突，在map中首先添加key值‘0’，索引值置为-1，这样在最后整除时再次添加key值0，发现有重复，判断一下索引差值即可正确判断返回值。

具体代码如下：

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) {
        int len = nums.size();
        if(k < 0)
            k = -k;
        if(len < 2)
            return false;
        map<int,int> mp;
        mp[0] = -1;  // 处理数组和刚好能整除k的情况
        int rem = 0;
        for(int i = 0;i < len;i++){
            rem += nums[i];
            if(k) // 避免除数为0
                rem = rem % k;
            if(mp.find(rem) != mp.end() ){
                if(i - mp[rem] > 1)
                    return true;                
            }
            else
                mp[rem] = i;
                       
        }
        return false;
    }
};