地址:https://leetcode-cn.com/problems/subarray-sum-equals-k/
思路:因为是连续子数组,因此比较容易想到“前缀和数组”。
方法一:前缀和数组
Java 代码:
public class Solution {
public int subarraySum(int[] nums, int k) {
int len = nums.length;
int[] preSum = new int[len + 1];
preSum[0] = 0;
for (int i = 0; i < len; i++) {
preSum[i + 1] = preSum[i] + nums[i];
}
int res = 0;
for (int left = 0; left < len; left++) {
for (int right = left + 1; right <= len; right++) {
if (preSum[right] - preSum[left] == k) {
res++;
}
}
}
return res;
}
}
方法二:哈希表
Java 代码:
import java.util.HashMap;
import java.util.Map;
public class Solution {
public int subarraySum(int[] nums, int k) {
Map<Integer, Integer> hash = new HashMap<>();
// 剩余值为 0 的时候设置为 1 ,这是因为 1 是乘法的单位元
hash.put(0, 1);
int sum = 0;
int res = 0;
for (int num : nums) {
sum += num;
if (hash.containsKey(sum - k)) {
res += hash.get(sum - k);
}
hash.put(sum, hash.getOrDefault(sum, 0) + 1);
}
return res;
}
}