【C++】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<cstdio>

using namespace std;

const int N = 100010;
typedef long long LL;
int n , k ;

LL s[N],cnt[N]; //s[N]是前缀和数组，cnt[N]为余数为i的数有多少个

int main(){
    scanf("%d%d",&n,&k);
    //读入n个数
    for(int i = 1; i <= n; i ++){
        scanf("%lld",&s[i]);
        s[i] += s[i - 1];
    }
    
    //答案
    LL res = 0;
    cnt[0] = 1; //余数为0的数有 0 一个
    for(int i = 1; i <= n ; i++){
        res += cnt[s[i] % k];
        cnt[s[i] % k ] ++;
    }
    
    printf("%lld\n",res);
    return 0;
}