[220210] Subarray Sum Equals K

Given an array of integers nums and an integer k, return the total number of continuous subarrays whose sum equals to k.

class Solution:
    def subarraySum(self, nums, k):

        sum, res = 0, 0
        # d = {sum : frequency}
        d = {0:1}

        for n in nums:
            # continuous sum
            sum += n
            if sum - k in d:
                res += d[sum - k]

            d[sum] = d.get(sum, 0) + 1

        return res