Subarrays Sum Divisible
Python
Hard
3 views
Problem Description
Read n integers and k. Count subarrays whose sum is divisible by k.
Input Format
First line n k. Second line n integers.
Output Format
One integer count.
Official Solution
import sys
p=sys.stdin.read().strip().split()
if len(p)<2: sys.exit(0)
n=int(p[0]); k=int(p[1])
a=list(map(int,p[2:2+n]))
mp={0:1}
sm=0
ans=0
for x in a:
sm+=x
r=sm%k
ans+=mp.get(r,0)
mp[r]=mp.get(r,0)+1
sys.stdout.write(str(ans))
Solutions (0)
No solutions submitted yet. Be the first!
No comments yet. Start the discussion!