20210602 每日一题 连续的子数组和

题目

题目链接

代码

class Solution {
public:
    bool checkSubarraySum(vector<int>& nums, int k) {

    }
};

方法一：前缀和+哈希表

分析

朴素思路：当数组 $nums$ 的长度为 $n$ 时，使用 $O(n^2)$的时间遍历遍历数组 $nums$ 的全部子数组，对于每个子数组需要 $O(n)$ 的时间计算元素和，因此时间复杂度是 $O(n^3)$。

同余定理：如果两个整数 $a$、$b$ 满足 $a-b$ 能被 $k$ 整除，那么$a$、$b$ 对 $k$ 同余。因此

因此我们首先计算出数组 $nums$ 的前缀和数组 $sums$：
$$sums[i]=\sum _{i=0}^{n}nums[i]$$ 假设 $0\le j<i<n$, 对于从下标 $j$ 到下标 $i$ 的子数组的元素和为$sums[i]-sums[j]$。根据同余定理从下标 $j$ 到下标 $i$ 的子数组的元素和为 $k$ 的倍数时，$sums[i]$ 和 $sums[j]$ 余数相同。因此，可以将题目表示为：

• 存在 $j$ 和 $i$ 使得 $sums[i]$ 和 $sums[j]$ 余数相同且$i-j\ge 2$时，返回 $true$；
• 存在 $j$ 和 $i$ 使得 $sums[i]$ 和 $sums[j]$ 余数相同且$i-j<2$时，不做操作继续运行；

在此基础上，我们可以使用哈希表记录任意余数值第一次出现的位置，并且使用一个动态更新数值 $cur$ 来替代前缀和数组 $sums$。因此，可以将题目进行简化：

• 对于第 $i$ 个当前余数：$cur=(cur+nums[i])\%k$；
• 如果当前余数在哈希表中不存在，则将当前余数和当前下标 $i$ 的以及当前余数 $cur$ 的键值对存入哈希表中。
• 如果当前余数在哈希表中已经存在，则取出该余数在哈希表中对应的下标 $j$，当$i-j\ge 2$时，返回 $true$；
• 如果当前余数在哈希表中已经存在且$i-j<2$时，不做操作继续运行；

代码

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

要点

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是数组 $nums$ 的长度。需要遍历数组一次。哈希表可以在 $O(1)$ 的时间进行查找。
• 空间复杂度：$O(min(n,k))$。空间复杂度主要取决于哈希表，哈希表中存储每个余数第一次出现的下标，最多有 $min(n,k)$ 个余数。

拓展