Leetcode560：和为K的子数组

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

思路：
这个题目是Leetcode1074:元素和为目标值的子矩阵数量的简化版。之前的题目是二维数组求和子集，现在变成了一维数组求子集。相当于略去了对上下边界的遍历，只对左右进行遍历搜索。
方法1： 就是上述思路展开的代码，前缀和+哈希表，基本和Leetcode1074的哈希表子部分一致。

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
    int len=nums.size();
    vector<int>sum(len+1);
    unordered_map<int,int>map;
    int count=0;
    map[0]=1; //很关键！
    for(int i=1;i<=len;i++){
       sum[i]=sum[i-1]+nums[i-1];
       
       if(map.find(sum[i]-k)!=map.end())
         count+=map[sum[i]-k]; //也就是说这个操作只对当前sum之前的元素进行检索。到它为止的检索依靠mp[0]=1；
        //上面的操作是看当前的sum之前是否存在一个结点，让这个结点到sum当前等于k！有的话就看看这个结点个数
        //结束之后就放入这个结点。原因在上面说了，单个结点是不是解可以判断，到这个结点为止是的话就是0，前面初始化也给出了。
         map[sum[i]]++; //这个一定要最后入哈希表，因为前面的计数中假设K=0，那每次都要+1了
    }
    return count;    
    }
};

对代码简化下就是：

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        unordered_map<int, int> mp;
        mp[0] = 1;
        int count = 0, pre = 0;
        for (auto& x:nums) {
            pre += x;
            if (mp.find(pre - k) != mp.end()) {
                count += mp[pre - k];
            }
            mp[pre]++;
        }
        return count;
    }
};

方法2： 枚举的思路，考虑以 ii结尾和为 kk的连续子数组个数，我们需要统计符合条件的下标 j 的个数。sum用来记录每次以i结尾的情况下，往前面走的子集之和，反正是递增的嘛~

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int count = 0;
        for (int start = 0; start < nums.size(); ++start) {
            int sum = 0;
            for (int end = start; end >= 0; --end) {
                sum += nums[end];
                if (sum == k) {
                    count++;
                }
            }
        }
        return count;
    }
};

这里顺便附上两数之和的代码：

class Solution {
public:
    vector<int> twoSum(vector<int>& nums, int target) {
        unordered_map<int, int> hashtable;
        for (int i = 0; i < nums.size(); ++i) {
            auto it = hashtable.find(target - nums[i]); //如果在哈希表中找到当前元素互补项就返回哈希表下标
            if (it != hashtable.end()) {
                return {it->second, i};
            }
            hashtable[nums[i]] = i;
        }
        return {};
    }
};