剑指 Offer II 010. 和为 k 的子数组
给定一个整数数组和一个整数 k ,请找到该数组中和为 k 的连续子数组的个数。
示例 1:
输入:nums = [1,1,1], k = 2
输出: 2
解释: 此题 [1,1] 与 [1,1] 为两种不同的情况
示例 2:
输入:nums = [1,2,3], k = 3
输出: 2
提示:
1 <= nums.length <= 2 * 104
-1000 <= nums[i] <= 1000
-107 <= k <= 107
思路
数组:0…i…j…end
当我们去想要判断第j个数是否有成立的条件时,我们只需要去判断第j个数减去0到i个数的和(0<i<j)是否等于k即可判断有无成立条件(区间i+1到j的和)。
所以我们只要吧前i个数的和记录下来,作为前缀,然后再用查找效率相对较高的hashmap进行记录前缀,即可快速找出答案。
class Solution {
public int subarraySum(int[] nums, int k) {
HashMap<Integer,Integer> map = new HashMap<>();
int pre = 0,count = 0;
map.put(0,1);
for(int i=0;i<nums.length;i++){
pre += nums[i];
if(map.containsKey(pre-k)){
count += map.get(pre-k);
}
//判断是否存在per键,若存在,则取pre值+1,若不存在,则创建一个,值为0
map.put(pre,map.getOrDefault(pre,0)+1);
}
return count;
}
}
代码为官方答案,此处记录个人理解思路
最后
以上就是留胡子小懒虫最近收集整理的关于高频面试题——和为 k 的子数组的全部内容,更多相关高频面试题——和为内容请搜索靠谱客的其他文章。
本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
发表评论 取消回复