restructure zeta/** to storage/zeta/**

This commit is contained in:
2025-05-10 21:54:24 +09:00
parent 2886820691
commit 2f2e0759fd
407 changed files with 7 additions and 1 deletions

View File

@@ -0,0 +1,17 @@
import sys
import math
input = sys.stdin.readline
if __name__ == "__main__":
N, M = map(int, input().split())
A = list(map(int, input().split()))
S = [(0, 0)]
C = [0] * 1001
C[0] = 1
for i in range(N):
s = ((S[-1][0] + A[i]) % M, i + 1)
S.append(s)
C[s[0]] += 1
print(sum([math.comb(c, 2) for c in C]))