题解一:
前缀和数组+哈希表:可以计算所有子数组之和暴力求解,但复杂度太高。对于子数组求和的过程,我们可以采用前缀和数组进行优化,前缀和数组中pre[index]代表nums[0]~nusm[index]之和,当我们要计算子数组nums[x]~nums[y]之和时,可以采用pre[y]-pre[x-1]来作差求和。
题目要求和为k的子数组,也就是pre[y]-pre[x-1]=k,可以转换成pre[y]-k=pre[x-1],我们用HashMap哈希表来存储pre[ ]及该前缀和的值当前出现的次数,我们遍历到pre[y]时,判断pre[x-1]在前面出现过几次,也就是以y结尾的满足条件的子数组有几组。
import java.util.HashMap;
import java.util.Map;
class Solution {
public int subarraySum(int[] nums, int k) {
Map<Integer, Integer> map = new HashMap<>();
int len = nums.length;
int result = 0;
map.put(0, 1);
int temp = 0;
for (int i = 0; i < len; i++) {
temp += nums[i];
if (map.containsKey(temp - k)) result += map.get(temp - k);
map.put(temp, map.getOrDefault(temp, 0) + 1);
}
return result;
}
}