LeetCode560. 和为K的子数组

题目

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

思路

• 暴力两层for循环，外层起点，内层数组尾部，遍历到和为k结果++。
• 前缀和+哈希表。前缀和的目的在于：当寻找到某个位置的前缀和设为preSum，在前缀和中存在preSum-k这个前缀和，即preSum-k这个前缀和的位置到preSum这个前缀和的位置之前的元素和为k。而前缀和可能出现相同的值，所以用哈希表Map<前缀和，前缀和出现的次数>来存储。

class Solution {
   public int subarraySum(int[] nums, int k) {
       int res=0,len=nums.length;
       int preSum=0;
       Map<Integer,Integer>map=new HashMap<>();
       /*这里需要put(0,1)是因为如果出现了某个值等于k，那么 preSum-k 即为0，value值需要为1，
       因为一个元素也是一个结果
       */
       map.put(0,1);
       for(int i=0;i<len;i++){
           preSum+=nums[i];
           res+=map.getOrDefault(preSum-k,0);
           map.put(preSum,map.getOrDefault(preSum,0)+1);
       }
       return res;
   }
}