leetcode 和为 k 的子数组

最新推荐文章于 2022-08-21 12:56:09 发布

酱酱熊

最新推荐文章于 2022-08-21 12:56:09 发布

阅读量361

点赞数

分类专栏：算法文章标签： leetcode 哈希算法算法

本文链接：https://blog.csdn.net/weixin_42496727/article/details/121253034

版权

算法专栏收录该内容

136 篇文章 1 订阅

订阅专栏

本文介绍了如何使用前缀和和哈希表的方法解决求解数组中和等于特定值k的子数组个数问题。通过遍历数组，利用哈希表记录每个元素出现的次数及其与k的差值，实现高效查找和更新。代码展示了如何利用这些技术实现Solution类的subarraySum方法。

摘要由CSDN通过智能技术生成

题目链接
思路：前缀和+hash表
每次去map中找前面是否出现过当前元素所需要的值（也就是当前元素距离k的值）。
加上所需要的值出现的次数（这个次数可能是0次）。
然后再把自己给放进map中。次数要加1
注意，mao中要提前放入0，次数是1，因为当前元素如果等于k，那么当前元素和k就相差0，所以0默认出现了1次。
代码：

class Solution {
    public int subarraySum(int[] nums, int k) {
        HashMap<Integer, Integer> map = new HashMap<>();
        int res = 0;

		//前缀和
        for(int i = 1; i < nums.length;i++){
            nums[i] = nums[i] + nums[i-1];
        }

        map.put(0,1);
        for(int i = 0;i < nums.length;i++){
            int count = map.getOrDefault(nums[i]-k,0);
            res = res + count;
            map.put(nums[i],map.getOrDefault(nums[i],0)+1);
        }
        return res;
    }
}