只对了一个,求大佬救救
查看原帖
只对了一个,求大佬救救
611860
hwqdsg楼主2024/3/21 17:57
import math
import statistics

n,k = list(map(int,input().split()))

lst = []
for i in range(n):
    x = int(input())
    lst.append(x)
Sum = [0 for i in range(100002)]

Sum[0] = lst[0]
for i in range(1,len(lst)):
    Sum[i] = Sum[i-1] + lst[i]


count = 0


m = [0 for i in range(k)]
for i in range(len(lst)):
    m[Sum[i]%k] += 1
    if lst[i]%k == 0:
        count += 1


for j in range(len(m)):
    count += int(m[j]*(m[j]-1)/2)

print(count)
2024/3/21 17:57
加载中...