Leetcode-560. 和为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]。

思路1：

    用hash表cul存放从第0个位置到第i个位置上的和出现的情况。sum记录数组到第i个位置上的和，如果sum-k在cul中出现过，说明从某个位置开始第i个位置存在和为k 的情况，因此res++；

c++ code：

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int sum = 0, res = 0;
        unordered_map<int, int> cul;
        cul[0] = 1;
        for (auto m : nums) {
            sum += m;
            res += cul[sum - k];
            cul[sum]++;
        }
        return res;
    }
};

思路2：

    创建一个数组sum，其中sum第i个位置存放的是nums中0到i-1位置上的和。因此可以在sum上通过两层循环，计算出和为k值的个数。

c++ code：

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int n = nums.size();
        if(n == 0){
            return 0;
        }
        int res = 0;
        vector<int> sum(n + 1, 0);
        for(int i = 1; i <= n; i++){
            sum[i] = nums[i - 1] + sum[i - 1];
        }
        for(int i = 0; i <= n; i++){
            for(int j = i + 1; j <= n; j++){
                if(sum[j] - sum[i] == k){
                    res++;
                }
            }
        }
        return res;
        
    }
};

思路3：

    在思路1的基础上进行修改，也可以不创建数组，只用sum记录从第i个位置到之后位置上的和，若sum==k，res++。

c++ code：

class Solution {
public:
    int subarraySum(vector<int>& nums, int k) {
        int n = nums.size();
        if(n == 0){
            return 0;
        }
        int res = 0;
        int sum;
        for(int i = 0; i < n; i++){
            sum = 0;
            for(int j = i; j < n; j++){
                sum += nums[j];
                if(sum == k){
                    res++;
                }
            }
        }
        return res;
        
    }
};