LeetCode——第二十六天（和为K的子数组、前缀和）

LeetCode——第二十六天

560. 和为K的子数组

给定一个整数数组和一个整数 k，你需要找到该数组中和为 k 的连续的子数组的个数。

示例 1 :

输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
说明 :

数组的长度为 [1, 20,000]。
数组中元素的范围是 [-1000, 1000] ，且整数 k 的范围是 [-1e7, 1e7]。

分析：这题目很熟悉有没有，很像之前一题前缀和，不过我是用了两个for循环做的，看了题解才知道前缀和的做法。这题目看着简单，不过有负数还有0的一些特殊情况需要考虑，用分治等是没办法的。

暴力：

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int res=0;
        //遍历
        for(int i=0;i<nums.size();i++){
            int sum=0;
            //从当前位置往前加，如果符合则res++
            for(int j=i;j>=0;j--){
                sum+=nums[j];
                if(sum==k) res++;
            }
        }
        return res;
    }
};

忘记之前介绍前缀和是哪个了，在提下好了，这边用LeetCode官方的文档。

我就从pre[j]=pre[i]-k这里开始说吧，前面要是不懂可以留言或者私信。像文档提到的我们考虑以i为结尾的和为k的连续子数组个数只要统计前缀和为pre[i]-k的pre[j]即可，因为pre数组是唯一的，那么统计其个数那就是子数组的个数了。

再建立一个hash表来记录次数，这样只要一遍更新就可以得到结果。

前缀和

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int,int> hash;        
        hash[0] = 1;    //空数组的前缀和为0，所以前缀和为0的连续子数组个数初始化为1
        int sum = 0,res = 0;
        for (int num : nums) {
            sum += num;
            if(hash.find(sum-k)!=hash.end()) res+=hash[sum-k];   //加上前缀和为sum-k的数组的个数
            hash[sum]++;    //前缀和为sum的个数++，这条语句要写在后面，写在前面有可能自己用到了自己。
        }
        return res;
    }
};

2.总结