K倍区间h

K倍区间

题目

给定一个长度为 N 的数列，A₁,A₂,…A ，如果其中一段连续的子序列 A_i,A_i+1,…A 之和是 K 的倍数，我们就称这个区间 [ i , j ] 是 K 倍区间。

你能求出数列中总共有多少个 K 倍区间吗？

输入格式

第一行包含两个整数 N 和 K 。

以下 N 行每行包含一个整数 A_i 。

输出格式

输出一个整数，代表 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;
}