K倍区间
题目
给定一个长度为 N 的数列,A1,A2,…A ,如果其中一段连续的子序列 Ai,Ai+1,…A 之和是 K 的倍数,我们就称这个区间 [ i , j ] 是 K 倍区间。
你能求出数列中总共有多少个 K 倍区间吗?
输入格式
第一行包含两个整数 N 和 K 。
以下 N 行每行包含一个整数 Ai 。
输出格式
输出一个整数,代表 K 倍区间的数目。
数据范围
1 ≤ N , K ≤ 100000 ,
1 ≤ A~i ~≤ 100000
输入样例:
5 2
1
2
3
4
5c
输出样例:
6
题意
给定一段数列和一个整数K,假若其中的一段数之和是K的倍数,则称这段区间为K倍区间,问这段数列有多少个K倍区间
思路
前缀和
摘抄一位大佬的题解(我好菜)
AC代码
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<cstring>
using namespace std;
typedef long long LL;
const int N = 100010;
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];
cnt[s[i] % k] ++ ;
}
printf("%lld\n", res);
return 0;
}