add shebang to run.py and complete 10986.py 11659.py 11660.py

This commit is contained in:
2024-09-09 19:19:16 +09:00
parent ec3bd95de4
commit bd8d2defaa
4 changed files with 47 additions and 0 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]))