算法系列——和为 K 的子数组(Subarray Sum Equals K)
Posted BridgeGeorge
tags:
篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了算法系列——和为 K 的子数组(Subarray Sum Equals K)相关的知识,希望对你有一定的参考价值。
题目
给你一个整数数组 nums 和一个整数 k ,请你统计并返回该数组中和为 k 的连续子数组的个数。
示例 1:
输入:nums = [1,1,1], k = 2
输出:2
示例 2:
输入:nums = [1,2,3], k = 3
输出:2
解答
哈希表+前缀和 时间复杂度为O(n),空间复杂度O(n)
class Solution
public int subarraySum(int[] nums, int k)
//前缀和
// pre[i] [0,i]的和
//[i,j] pre[j]-pre[i-1]=k
int sum=0;
int count=0;
Map<Integer,Integer> map=new HashMap<>();
map.put(0,1);//此处必须赋值为1
for(int i=0;i<nums.length;i++)
sum+=nums[i];
if(map.containsKey(sum-k))
count+=map.get(sum-k);
map.put(sum,map.getOrDefault(sum,0)+1);
return count;
以上是关于算法系列——和为 K 的子数组(Subarray Sum Equals K)的主要内容,如果未能解决你的问题,请参考以下文章