AtCoder ABC 105 D
python に慣れるため久しぶりにプロコンをやってみた。今の AtCoder になってから初だと思う。
D: Candy Distribution - AtCoder Beginner Contest 105 | AtCoder
code
from collections import defaultdict N, M = map(int, input().split()) A = list(map(int, input().split())) Amod = defaultdict(int) sum_mod = 0 for v in A: sum_mod = (sum_mod + v) % M Amod[sum_mod] += 1 sum_mod = 0 ans = 0 for v in A: ans += Amod[sum_mod] sum_mod = (sum_mod + v) % M Amod[sum_mod] -= 1 print(ans)