题目描述：
给定一个长度为 NN 的数列，A1,A2,…AN，如果其中一段连续的子序列 Ai,Ai+1,…Aj 之和是 K 的倍数，我们就称这个区间 [i,j] 是 K 倍区间。

你能求出数列中总共有多少个 K 倍区间吗？

输入格式

第一行包含两个整数 N 和 K。

以下 NN 行每行包含一个整数 Ai。

输出格式

输出一个整数，代表 K 倍区间的数目。

数据范围

1≤N,K≤100000,
1≤Ai≤100000

输入样例：

5 2
1
2
3
4
5

输出：

6

import os
import sys

# 请在此输入您的代码
# 前缀和

n,k = map(int,input().split())
res = 0
s = [0] #存前缀和
cnt = [0 for i in range(k)] #存区间和模k的余数。元素的初值为0
cnt[0] = 1 #当单独的某一个数模k==0也是一种情况，所以cnt[0]开始需要+1



#前缀和余数为0（从头开始） 或者 余数相同的（切片） 都满足k倍区间
for i in range(n):#range(n)表示[0,n)
  s.append(int(input()))
for i in range(1,n+1):# 预处理前缀和
  s[i] +=s[i-1]
for i in range(1,n+1):
  res += cnt[s[i]%k]
  cnt[s[i]%k] +=1
print(res)

参考