0
点赞
收藏
分享

微信扫一扫

面试题10:和为k的子数组(Java版)

婉殇成长笔记 2022-05-01 阅读 28

题目:输入一个整数数组和一个整数k,请问数组中有多少个数
字之和等于k的连续子数组?例如,输入数组[1,1,1],k的值为
2,有2个连续子数组之和等于2。

// 因为求得是一个连续子数组,所以当m大于n时,第前m项的和减去
// 第前n项的和等于k则代表n+1到m之间连续的数组的和等于k以此来
// 求出和为k的连续数字的个数
public static int subArraySum(int[] nums, int k) {
    Map<Integer, Integer> sumToCount = new HashMap<>();
    // 当sum - k等于0时说明当前连续数字本身就满足条件,
    // 所以需要初始哈希表中key为0的value为1
    sumToCount.put(0, 1);
    int sum = 0;
    int count = 0;
    for (int num: nums) {
        sum += num;
        // count加上key为sum - k的键在哈希表中对应的value
        // (即加上了连续数组之和为sum - k的个数)
        count += sumToCount.getOrDefault(sum - k, 0);
        // 将哈希表中对应和的数量加一
        sumToCount.put(sum, sumToCount.getOrDefault(sum, 0) + 1);
    }
    return count;
}
举报

相关推荐

0 条评论