给定一个长度为 N 的数列,A1,A2,…AN,如果其中一段连续的子序列 Ai,Ai+1,…Aj 之和是 K 的倍数,我们就称这个区间 [i,j] 是 K 倍区间。
你能求出数列中总共有多少个 K 倍区间吗?
输入格式
第一行包含两个整数 N 和 K。
以下 N 行每行包含一个整数 Ai。
输出格式
输出一个整数,代表 K 倍区间的数目。
数据范围
1≤N,K≤100000,
1≤Ai≤100000
输入样例:
5 2
1
2
3
4
5
输出样例:
6
cpp
#include<bits/stdc++.h> // 引入所有标准库,包括输入输出流、字符串处理等
using namespace std; // 使用标准命名空间
int n, k; // 定义整型变量n和k,分别用于存储数列的长度和倍数K
const int N = 100010; // 定义常量N,作为数组和循环的上界
int a[N], s[N]; // 定义整型数组a和s,分别用于存储数列和前缀和
typedef long long ll; // 使用typedef为long long定义一个简写别名ll
int main(){
scanf("%d%d", &n, &k); // 从标准输入读取两个整数n和k
for(int i = 1; i <= n; i++){ // 循环n次,从1到n
scanf("%d", &a[i]); // 从标准输入读取数列的第i个元素
s[i] = s[i - 1] + a[i]; // 计算前缀和,s[i]存储从第一个元素到第i个元素的和
}
ll res = 0; // 定义一个长整型变量res,用于存储K倍区间的数量
for(int r = 1; r <= n; r++){ // 外层循环,r代表区间的右边界
for(int l = 1; l <= r; l++){ // 内层循环,l代表区间的左边界
int sum = s[r] - s[l - 1]; // 计算区间[l, r]的和
if(sum % k == 0){ // 如果区间和是K的倍数
res++; // K倍区间数量加1
}
}
}
cout << res; // 输出K倍区间的数量
return 0; // 程序结束,返回0
}