历届试题 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

本题需要考利用数组前缀和来降低算法复杂度

前缀和：数组a[i]=a[0]+...+a[i];

for(int i = 1;i < n;i++)
{
a[i] = a[i] + a[i-1];
}

#include<iostream>
#include<cstring>
using namespace std;

int main()
{
	int a[1000010];
	int b[1000010];
	long long int sum = 0;
	memset(b,0,sizeof(b));
	int n;
	long long int m;
	cin>>n>>m;
	for(int i = 0;i < n;i++)
	{
		cin>>a[i];	
	}	
	a[0] = a[0] % m;
	for(int i = 1;i < n;i++)
	{
		a[i] = (a[i] + a[i-1]) % m;
	}
	for(int i = 0;i < n;i++)
	{
		sum += (b[a[i]]++); 
	}
	cout<<sum+b[0];
	return 0;
	
}