560. 和为K的子数组

560. 和为K的子数组

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

示例 1 :

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sum-equals-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

基本思路：暴力求解，时间复杂度为O(n2)，超时

    int subarraySum(vector<int>& nums, int k) {
        vector<int> sum(nums.size()+1,0);
        for(int i=0;i<nums.size();i++){
            sum[i+1]=sum[i]+nums[i];
        }
        int cnt=0;
        for(int i=0;i<nums.size();i++){
            for(int j=i+1;j<=nums.size();j++){
                int tmp=sum[j]-sum[i];
                if(k==tmp)
                    cnt++;

            }
        }

        return cnt;


    }

基本思路： 利用哈希表map，来存储前缀和的个数。

• i~j的和：preSum[j]-preSum[i-1]=k，即当确定j,真正决定以j结尾的连续数组的个数为前面和为preSum[j]-k的个数，故可以采用map存储前缀和的个数;
• preSum[i-1]=preSum[j]-k，若preSum[i-1]存在，则cnt+=mp[preSum[i-1];

    int subarraySum(vector<int>& nums, int k) {
        int cnt=0,sum=0;
        unordered_map<int,int> m;
        m[0]=1; //表示sum==k;
        for(int i=0;i<nums.size();i++){
           sum+=nums[i];
           if(m.find(sum-k)!=m.end()){
               cnt+=m[sum-k];
           }
           m[sum]++;
        }

        return cnt;


    }

类似：437. 路径总和 III