给定一个长度为 NN 的数列,A1,A2,…ANA1,A2,…AN,如果其中一段连续的子序列 Ai,Ai+1,…AjAi,Ai+1,…Aj 之和是 KK 的倍数,我们就称这个区间 [i,j][i,j] 是 KK 倍区间。
你能求出数列中总共有多少个 KK 倍区间吗?
输入格式
第一行包含两个整数 NN 和 KK。
以下 NN 行每行包含一个整数 AiAi。
输出格式
输出一个整数,代表 KK 倍区间的数目。
数据范围
1≤N,K≤1000001≤N,K≤100000,
1≤Ai≤1000001≤Ai≤100000
输入样例:
5 2
1
2
3
4
5
输出样例:6
题解:
个人认为最难理解的就是cnt[0]=1(记录余数为0的数的个数)这一步,这么初始化就很细节,通常我们的前缀和s[0]没有意义,但是对于此题来说,优化到最后就是找与s[R]%k余数相同的s[l],然后累加,但是记录余数为i的数组cnt[0]序列,我们发现当余数为0时还有一种情况就是他自己本身也可以整除k,所以我们需要每次加上这个序列的同时也加上它本身情况(+1)
AC代码:
#include<stdio.h>
#include<iostream>
#include<algorithm>
typedef long long ll;
const int N=100010;
using namespace std;
int n,k;
ll s[N],cnt[N];
int main()
{
scanf("%d%d",&n,&k);
for(int i=1;i<=n;i++)
{
scanf("%lld",&s[i]);
s[i]+=s[i-1];//处理前缀和
}
ll res=0;
cnt[0]=1;//看题解
for(int i=1;i<=n;i++)
{
res+=cnt[s[i]%k];//找与s[R]%k余数相同的s[l]
cnt[s[i]%k]++;//至少需要两个起步才可以算开始基数
}//但是对于余数为0的话不需要大于2才开始计数,因为它本身可以作为一个序列整除
printf("%d\n",res);
return 0;
}