|
给定一个长度为 N?的数列，A1,A2,…AN，如果其中一段连续的子序列 Ai,Ai+1,…Aj 之和是 KK 的倍数，我们就称这个区间 [i,j] 是 K?倍区间。
你能求出数列中总共有多少个 K 倍区间吗？
输入格式
第一行包含两个整数 N和 K。
以下 N?行每行包含一个整数 Ai。
输出格式
输出一个整数，代表 K?倍区间的数目。
数据范围
1≤N,K≤100000,
1≤Ai≤100000
输入样例：
5 2
1
2
3
4
5
输出样例：
6
#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long LL;
const int N = 100010;
LL num[N] , yus[N];
int main()
{
int n , k ;
LL ans = 0;
scanf("%d%d", &n, &k);
for (int i = 1; i <= n; i ++ ){
LL x;
cin >> x;
num[i] = num[i - 1] + x;
}
LL res = 0;
yus[0] ++;
for (int i = 1; i <= n; i ++ ){
ans += yus[num[i] % k];
yus[num[i] % k] ++;
}
cout << ans << endl;
return 0;
}
?
|