leetcode 560. 和为K的子数组

leetcode 560. 和为K的子数组

题目详情

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

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

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

我的代码

class Solution {
private:
    vector<int> array;
    int len;
    int loc;
public:
    inline int getRes(int start, int k) {
        if ((k == 0) && (start > loc)) {
            return start >= len? 1: getRes(start + 1, k - array[start]) + 1;
        }
        if (start >= len) {
            return 0;
        }
        return getRes(start + 1, k - array[start]);
    }

    int subarraySum(vector<int>& nums, int k) {
        array = nums;
        len = nums.size();
        int res = 0;
        for (int i = 0; i < len; ++i) {
            loc = i;
            res += getRes(i, k);
        }
        return res;
    }
};

我的成绩

执行结果：通过
执行用时 : 1828 ms, 在所有 C++ 提交中击败了5.10%的用户
内存消耗 : 16.8 MB, 在所有 C++ 提交中击败了33.33%的用户

一些想法

本道题我使用了递归，但并没有进行减枝，效率比较低下。需要特别注意的是当k = 0时的情况。

执行用时为 20 ms 的范例

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

思考

没看懂范例。。。