剑指 Offer II 010. 和为 k 的子数组(Java)

最新推荐文章于 2024-07-24 20:44:05 发布

侯一鸣Supermonkey

最新推荐文章于 2024-07-24 20:44:05 发布

阅读量115

点赞数

分类专栏： java后端学习笔记文章标签：算法数据结构 c++

本文链接：https://blog.csdn.net/qq_19830591/article/details/127760129

版权

java后端学习笔记专栏收录该内容

61 篇文章 0 订阅

订阅专栏

O(n^2)

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

O(n)

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