当前位置 博文首页 > -happy 的博客:LeetCode - 哈希表 -560. 和为K的子数组
难度 中等
给定一个整数数组和一个整数 k,你需要找到该数组中和为 k 的连续的子数组的个数。
示例1:
输入:nums = [1,1,1], k = 2
输出: 2 , [1,1] 与 [1,1] 为两种不同的情况。
枚举法,
枚举所有的子数组,然后求和判断是否为 k,这种解法复杂度达到了 n ^ 3 所以不可取,但是可以稍微优化一下, 不需要遍历子数组,可以直接根据前一个的值计算部分和,省去了一个 n 的循环,复杂度降为 n ^ 2
前缀和 + 哈希表优化:
这个方法我弄的还不是很清楚。
这里给个解法链接
https://leetcode-cn.com/problems/subarray-sum-equals-k/solution/he-wei-kde-zi-shu-zu-by-leetcode-solution/
枚举法只需要数组即可。
枚举法:
二重循环,加求部分和,判断是否等于 k 即可。
实现代码(java):
枚举法:
public class Solution {
public int subarraySum(int[] nums, int k) {
int count = 0;
for (int start = 0; start < nums.length; ++start) {
int sum = 0;
for (int end = start; end >= 0; --end) {
sum += nums[end];
if (sum == k) {
count++;
}
}
}
return count;
}
}
前缀和 + 哈希表优化:
public class Solution {
public int subarraySum(int[] nums, int k) {
// hash
// 记录合适的连续字符串数量
int count=0;
// 记录前面数字相加之和
int pre=0;
// map记录前几个数字之和为K出现相同和的次数为V
HashMap<Integer,Integer> map = new HashMap<>();
// 初始化
map.put(0,1);
for (int i = 0; i < nums.length; i++) {
pre+= nums[i];
// 如果前面数字之和加上这个数字正好等于K(存在一个数字加上nums[i]结果为K
// 说明找到了
if (map.containsKey(pre-k)){
// 累计
count+=map.get(pre-k);
}
// 计算新的和放入map
map.put(pre,map.getOrDefault(pre,0)+1);
}
return count;
}
}
cs