和为k的子数组
给你一个整数数组 nums 和一个整数 k ,请你统计并返回该数组中和为 k 的连续子数组的个数。
示例 1:
输入:nums = [1,1,1], k = 2
输出:2
(上述题目来源于LeetCode)
解法一(暴力解法1)
class Solution {
public int subarraySum(int[] nums, int k) {
//暴力解法
int count = 0;
int n = nums.length;
for(int i = 0;i < n;i++){
int result = 0;
for(int j = i;j < n;j++){
result = result + nums[j];
if(result == k){
count++;
}
}
}
return count;
}
}
解法二(暴力解法2)
class Solution {
public int subarraySum(int[] nums, int k) {
//枚举
int count = 0;
int n = nums.length;
for(int i = 0;i < n;i++){
int result = 0;
for(int j = i;j >= 0;j--){
result = result + nums[j];
if(result == k){
count++;
}
}
}
return count;
}
}
解法三(辅助数组)
class Solution {
public int subarraySum(int[] nums, int k) {
int count = 0;
int n = nums.length;
int[] sum1 = new int[n];
int[] sum2 = new int[n];
sum1[0] = nums[0];
sum2[0] = nums[0];
for(int i = 1;i < n;i++){
sum1[i] = sum1[i - 1] + nums[i];
sum2[i] = sum2[i - 1] + nums[i];
}
for(int i = 0;i < n;i++){
if(sum1[i] == k){
count++;
}
for(int j = i + 1;j < n;j++){
if(sum2[j] - sum1[i] == k){
count++;
}
}
}
return count;
}
}
解法四(前缀和+哈希表)
class Solution {
public int subarraySum(int[] nums, int k) {
//前缀和 + 哈希表
int count = 0;
int n = nums.length;
//将放置前缀和的数组长度置为n+1是因为要和下方新建的哈希表一一对应
int[] sum = new int[n + 1];
Map<Integer,Integer>hashmap = new HashMap<>();
//哈希表的第一位放(0,1)是因为方便找到前缀和等于k的情况
hashmap.put(0,1);
//计算前缀和
for(int i = 0;i < n;i++){
sum[i + 1] = sum[i] + nums[i];
//如果哈希表中存在key为sum[i] - k
if(hashmap.containsKey(sum[i + 1] - k)){
count = count + hashmap.get(sum[i + 1] - k);
}
//将当前的前缀和放入哈希表中
hashmap.put(sum[i + 1],hashmap.getOrDefault(sum[i + 1],0) + 1);
}
return count;
}
}