k倍区间（蓝桥杯真题）

给定一个长度为N的数列，A1, A2, ... AN，如果其中一段连续的子序列Ai, Ai+1, ... Aj(i <= j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。  

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

---

输入
-----
第一行包含两个整数N和K。(1 <= N, K <= 100000)  
以下N行每行包含一个整数Ai。(1 <= Ai <= 100000)  

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

---

例如，
输入：
5 2
1  
2  
3  
4  
5  

程序应该输出：
6

资源约定：
峰值内存消耗（含虚拟机） < 256M
CPU消耗  < 2000ms

---

Java代码：

import java.util.*;
public class Main{
	public static void main(String[] args){
		Scanner in=new Scanner(System.in);
		int n=in.nextInt(),k=in.nextInt();
		int[] a=new int[n];
		for(int i=0;i<n;i++)
			a[i]=in.nextInt();
		int count=0;
		for(int i=0;i<n;i++){//以a[i]开头
			int sum=0;
			for(int j=i;j<n;j++){//以a[j]结尾
				sum+=a[j];
				if(sum%k==0)
					count++;
			}
		}
		System.out.println(count);
	}
}

---

Python代码：

n,k=map(int,input().split())
a=[]  # 数列
for i in range(n):
    a.append(int(input()))
count=0
for i in range(n):  # 以 a[i] 开头
    sum=0
    for j in range(i,n):  # 以 a[j] 结尾
        sum+=a[j]
        if sum%k==0:
            count+=1
print(count)