Subarray Sum Equals K

问题描述:

Given an array of integers and an integer k, you need to find the total number of continuous subarrays whose sum equals to k.

Example 1:

Input:nums = [1,1,1], k = 2
Output: 2
思路:记录从开始到位置i的累加和,再统计位置i之前有多少个累加和等于sum[i]-k.
代码:class Solution {
public:
    int subarraySum(vector& nums, int k) {
        int n=nums.size();
        if (n==0)
            return 0;
        //计算到位置i的累加和
        vector sum(n+1,0);
        for (int i=1; i<=n; i++)
            sum[i]=nums[i-1]+sum[i-1];
        //查看在位置i之前,是否有sum[j]=sum[i]-k
        unordered_map hash;
        hash[0]=1;
        int res=0;
        for (int i=1; i<=n; i++)
        {
            if (hash[sum[i]-k])
                res+=hash[sum[i]-k];
            hash[sum[i]]++;
        }
        return res;
    }
};

你可能感兴趣的:(leetcode)