【算法挨揍日记】day15——560. 和为 K 的子数组、974. 和可被 K 整除的子数组

 【算法挨揍日记】day15——560. 和为 K 的子数组、974. 和可被 K 整除的子数组_第1张图片

 560. 和为 K 的子数组

560. 和为 K 的子数组

题目描述:

给你一个整数数组 nums 和一个整数 k ,请你统计并返回 该数组中和为 k 的连续子数组的个数 

子数组是数组中元素的连续非空序列。

 【算法挨揍日记】day15——560. 和为 K 的子数组、974. 和可被 K 整除的子数组_第2张图片

解题思路:

我们可以很容易想到暴力解法,但是时间复杂度为N^2,我们可以是用前缀和对其优化

 我们可以利用前缀和数组sum来记录,sum【i】代表到i位置的子数组之和

【算法挨揍日记】day15——560. 和为 K 的子数组、974. 和可被 K 整除的子数组_第3张图片

假设这是0-i的数组后面的我们先不看,我们可以将其分成两部分,一部分之和为k,另外一部分为sum【i】-k,本题是求和为k的数组的个数

那问题就可以变为在sum【i】-k中有多少个子数组等于sum【i】-k

这段区间正负都有,子区间可能不只有一个噢!!!

我们可以利用hash来完成本题,一个参数为前缀和,一个参数为次数 ,都是int类型

我们可以利用一个int变量来代替sum数组,因为sum不用每个都记录下来,只要记录上一个位置

解题代码:

class Solution {
public:
    int subarraySum(vector& nums, int k) {
        unordered_maphash;
        hash[0]=1;
        int ret=0;
        int sum=0;
        for(auto x:nums)
        {
            sum+=x;
            if(hash.count(sum-k))ret+=hash[sum-k];
            hash[sum]++;
        }
        return ret;
    }
};

 974. 和可被 K 整除的子数组

974. 和可被 K 整除的子数组

题目描述:

给定一个整数数组 nums 和一个整数 k ,返回其中元素之和可被 k 整除的(连续、非空) 子数组 的数目。

子数组 是数组的 连续 部分。

【算法挨揍日记】day15——560. 和为 K 的子数组、974. 和可被 K 整除的子数组_第4张图片

解题思路:

解决本题我们先来补充两个知识点 

  • 同余定理:(a-b)/p=k....0可以转换为a%p=b%p,具体证明可见同余定理_百度百科 (baidu.com)
  •  C++中,负数(a)%正数(p),在C++负数求余正数正确应该为正数,但是计算结果为负数,我们对其修正时期变成a%p+p,但是为了考虑到正负统一的问题,我们再次进行修正让其变为(a%p+p)%p

接下来我们来看一下本题,本题如果你做了上一题,你会发现基本上是类似的

只不过判断条件不太一样罢了

【算法挨揍日记】day15——560. 和为 K 的子数组、974. 和可被 K 整除的子数组_第5张图片

题目要求的可以被k整除的数组为图中sum-x部分,那就变成(sum-x)%k==0也就变成了sum%k==x%k,也就转为为在【0,i-1】这个区间内有多少个前缀和的余数等于sum%k

解题代码: 

class Solution {
public:
    int subarraysDivByK(vector& nums, int k) {
        unordered_maphash;
        hash[0%k]=1;
        int ret=0;
        int sum=0;
        for(auto x:nums)
        {
            sum+=x;
            int r=(sum%k+k)%k;
            if(hash.count(r))ret+=hash[r];
            hash[r]++;
        }
        return ret;
    }
};

你可能感兴趣的:(算法挨揍日记,Leetcode,数据结构)