《leetcode hot 100》560. 和为 K 的子数组

最新推荐文章于 2024-09-11 22:48:33 发布

gentle coder

最新推荐文章于 2024-09-11 22:48:33 发布

阅读量131

点赞数 3

文章标签：算法数据结构 java

本文链接：https://blog.csdn.net/m0_74969835/article/details/137001300

版权

思路：

1.前缀和+枚举（ON2）

2.前缀和+哈希（ON）

1：Acode:

class Solution {
    public int subarraySum(int[] nums, int k) {
        int[] pre = new int[nums.length];
        int ans=0;
        pre[0]=nums[0];
        if(pre[0]==k)ans++;
        for (int i = 1; i < nums.length; i++) {
            pre[i]=pre[i-1]+nums[i];
            if(pre[i]==k)ans++;
        }
        for (int i = 0; i < nums.length-1; i++) {
            for (int j = i+1; j < nums.length; j++) {
                if(pre[j]-pre[i]==k)ans++;
            }
        }
        return ans;
    }
}

public class Solution {
    public int subarraySum(int[] nums, int k) {
        int count = 0, pre = 0;
        HashMap < Integer, Integer > mp = new HashMap < > ();
        mp.put(0, 1);
        for (int i = 0; i < nums.length; i++) {
            pre += nums[i];
            if (mp.containsKey(pre - k)) {
                count += mp.get(pre - k);
            }
            mp.put(pre, mp.getOrDefault(pre, 0) + 1);
        }
        return count;
    }
}

over~