蓝桥杯：K倍区间（前缀和+取模优化 Python）

题目描述：

给定一个长度为 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

分析：

看到区间求和问题 第一反应自然是前缀和 但本题如果只用前缀和进行优化的话 时间复杂度仍然达到了O(n^2) 所以我们这里还需要优化 我们可以预先处理出一个前缀和数组 然后想到一个区间的和能整除k => 区间[l,r]的和是k的倍数即(sum[r] - sum[l-1])%k == 0 即sum[r]%k == sum[l-1]%k

然后我们在用一个数组cnt储存每个区间的和模k的余数个数 如cnt[1]就是余数为1的区间个数 当每枚举一个区间时 加上这个区间和模k的cnt即可。

想到这里 就可以写代码啦～

N = 100010
n,k = map(int,input().split())
lst = [0]
cnt = [0 for i in range(N)]
for i in range(n) : lst.append(int(input()))
for i in range(1,n+1) : lst[i] += lst[i-1] # 预处理前缀和
res = 0 
cnt[0] += 1 # 因为单独的某一个数模k==0也是一种情况，所以cnt[0]开始需要+1
for i in range(1,n+1) :
    res += cnt[lst[i]%k]
    cnt[lst[i]%k] += 1
print(res)