Acwing1230. K倍区间

给定一个长度为 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

#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;
}