思路:
这道题和之前的路径总和3是一样的,都是求从之前的某个节点到当前节点之间的连续路径的总和是否为某个值。所谓前缀和,就是在记录从路径最初的节点到当前节点的路径上的节点值之和。前缀路径和最适合解决的问题就是求有多少条连续路径总和为k。在一条路径上的先后两个节点之间的连续路径的总和等于后面节点的前缀和-前面节点的前缀和。
我们求从路径的起始节点到当前节点之间有多少条总和为k的路径,只需要求出当前节点之前,有多少个节点的前缀路径和等于当前节点的前缀路径和-k就可以了。
所以,我们通常都需要建立一个哈希表,键为前缀路径总和m,键值为前缀路径和为m的节点数量。
然后我们还需要设立一个变量sum来记录节点的前缀和。sum初值为0。
还需要设立一个count变量统计总和为k的路径总共有多少条。
然后我们每遍历到一个节点,就先用sum+当前节点值求出当前节点的路径和。然后我们去map中找之前节点前缀和为sum-k的数量n,n就是以当前节点结尾的总和为k的路径条数,将count+=n。然后我们将map中sum对应的键值加1。继续下一次遍历。
class Solution {
public int subarraySum(int[] nums, int k) {
HashMap<Integer,Integer> map = new HashMap<>();
int pre = 0;
int count = 0;
map.put(0,1);
for(int i=0;i<nums.length;i++)
{
pre += nums[i];
count += map.getOrDefault(pre-k,0);
int num = map.getOrDefault(pre,0);
map.put(pre,++num);
}
return count;
}
}