k倍区间-蓝桥杯

2017省赛蓝桥杯k倍区间python

最近在准备蓝桥杯，有个寒假刷题，自己写了个超时了，然后又找了找，终于找到一个没超时的。改成python代码后，看了半天没看懂，然后又去翻了翻，看了别人的评论终于弄懂了。写个博客记录一下。

思路大概就是：先算前n项和，并记录余数。然后现在从第一项开始的k倍区间都已经计数了，接着要算从第k项开始的k倍区间（1<k<=n)。k倍区间的余数必须是0，下面就要算前k个区间减去前面第j个区间的和为0的个数(1<=j<k)。因为区间是连续的，所以他们的余数必须相等。也就是count=count+$C_i^2$，i为余数相等的个数。
即代码里的count=count+flag[i]*(flag[i]-1)/2

import os
import sys
n,k=input().split()
n=int(n)
k=int(k)
flag=[0 for i  in range(1000000)]
x=[]
count=0
x.append(int(input()))
x[0]=x[0]%k
if x[0]==0:
    count += 1
flag[x[0]]=flag[x[0]]+1
for i in range(1,n):
    a=int(input())
    x.append((a%k+x[i-1])%k)
    flag[x[i]]=flag[x[i]]+1
    if x[i]==0:
        count += 1
for i in range(k):
    count=count+flag[i]*(flag[i]-1)/2
print(int(count))