leetcode523. 连续的子数组和

题目

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/continuous-subarray-sum

分析

这题范围是1E4,第一感觉是前缀和.但是感觉O(n^2)过不去,但是题解里有,
使用前缀和来遍历所有的区间,看有无区间符合条件
要注意k=0的情况,因为这时不能使用%,所以需要特判,sum[i,j]=0的时候,就返回true
O(n²)

代码

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

代码

这里使用了同余的原理,我是没想到.
对于(sum[i]-sum[j])%k==0,
-> sum[i]%k=sum[j]%k.所以我们可以在O(n)时间内进行计算.使用一个unordered_map来记录前面的sum[]%k的结果,后面查询是否有,并且查询出现的索引.之前没有出现过的就当前的索引的结果存入map里.妙啊!

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