class Solution:
def subarraysDivByK(self, nums: List[int], k: int) -> int:
cnt = [0 for _ in range(k)]
for i in range(len(nums)):
nums[i] = (nums[i] + nums[i-1]*(i>0) + 2*k) % k
cnt[nums[i]] += 1
ans = cnt[0]
for c in cnt:
ans += c * (c-1) // 2
return ans
백준 10986(골드3)과 거의 같은 문제.
어제 귀성하느라 문제를 못 풀었음;
그것도 대충 보니 subarray sum 관련이었는데 데일리가 좀 연관이 있게 나오는듯?
저번주는 tree만 존나 나왔던데
https://leetcode.com/problems/subarray-sums-divisible-by-k/