二刷560. Subarray Sum Equals K

Medium
一刷的笔记是直接复制粘贴过来的别人的语言,果然就完全没印象。虽然看似快,其实这样是最事倍功半的。一定要懂了,用自己的语言简单总结一下就好。别直接copy,没有意思。

第一种解法是O(N2)的brute force,就不写了。第二种方法是O(N)利用sum[i, j] = sum[0, j] - sum[0, i-1] where sum[i,j] stands for sum for substring[i,j] inclusively. 利用preSum这个装着preSum和它的频次的map, 遍历是一边加sum以便存入preSum.一般检查有没有之前的presum = sum – k的,为什么是sum – k的呢?因为preSum[0,i] + subarraySum = preSum[0, j], 知道 subarraySum = k, preSum[0, j] = sum, 所以要找有没有preSum[0,i] = sum – k的。能找到几个这样的preSum, 就能找到几个与之对应的subarray, 所以res++.

class Solution {
    public int subarraySum(int[] nums, int k) {
        if (nums == null || nums.length == 0){
            return 0;
        }
        int res = 0;
        int sum = 0;
        Map<Integer, Integer> preSum = new HashMap<>();
        preSum.put(0, 1);
        for (int i = 0; i < nums.length; i++){
            sum += nums[i];
            if (preSum.containsKey(sum - k)){
                res += preSum.get(sum - k);    
            }
            preSum.put(sum, preSum.getOrDefault(sum, 0) + 1);
        }
        return res;
    }
}
    原文作者:greatfulltime
    原文地址: https://www.jianshu.com/p/7edc49b9c6c4
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞