thiagowfx's avatar

¬ just serendipity 🍀 (not just serendipity)

ByteByteGo: K-Sum Subarrays

• 54 words • 1 min • updated

ByteByteGo: K-Sum Subarrays:

python
from typing import List
import itertools

def k_sum_subarrays(nums: List[int], k: int) -> int:
    acc = list(itertools.accumulate(nums))

    ans = 0

    for i in range(len(nums)):
        for j in range(i, len(nums)):
            mysum = acc[j] - (acc[i - 1] if i != 0 else 0)
            if mysum == k:
                ans += 1

    return ans