二刷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;
    }
}
?著作權歸作者所有,轉(zhuǎn)載或內(nèi)容合作請聯(lián)系作者
【社區(qū)內(nèi)容提示】社區(qū)部分內(nèi)容疑似由AI輔助生成,瀏覽時請結合常識與多方信息審慎甄別。
平臺聲明:文章內(nèi)容(如有圖片或視頻亦包括在內(nèi))由作者上傳并發(fā)布,文章內(nèi)容僅代表作者本人觀點,簡書系信息發(fā)布平臺,僅提供信息存儲服務。

相關閱讀更多精彩內(nèi)容

友情鏈接更多精彩內(nèi)容