蓝桥杯真题-- K倍区间（超详细，算法小白编写）

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

 

//由于是(total[r]-total[l-1])%k==0
//为了简化循环,直接将total[r]与total[l-1]分别当作前缀和其中的一个元素
//total[r]%k=total[l-1]%k就满足题意
//所以开一个哈希表统计前缀和%k相同的元素
//利用数学组合公式从每种方案中抽取两个配对n*(n-1)/2
//由于当余数为0时单个元素也属于k的倍数所以最后需要加上一次cnt[0]
//其中一点很重要本题数据量较大，如果及其恶劣所有数都满足，这样的方案数就是n+n*(n-1)/2
//这样一定爆int,所以最好使用long long

//暴力前缀和（AC不了）
// #include<iostream>

// using namespace std;

// const int N=1e5+10;

// int n,k;
// int temp[N],total[N],t;

// int main()
// {
//     scanf("%d%d",&n,&k);
//     for(int i=1;i<=n;i++)
//     {
//         scanf("%d",&temp[i]);
//         total[i]=total[i-1]+temp[i];
//     }
    
//     for(int i=1;i<=n;i++)
//         for(int j=i;j<=n;j++)
//         {
//             if((total[j]-total[i-1])%k==0) t++;
//         }
//     printf("%d",t);
//     return 0;
// }
//由于是(total[r]-total[l-1])%k==0
//为了简化循环,直接将total[r]与total[l-1]分别当作前缀和其中的一个元素
//total[r]%k=total[l-1]%k就满足题意
//所以开一个哈希表统计前缀和%k相同的元素
//利用数学组合公式从每种方案中抽取两个配对n*(n-1)/2
//由于当余数为0时单个元素也属于k的倍数所以最后需要加上一次cnt[0]
//其中一点很重要本题数据量较大，如果及其恶劣所有数都满足，这样的方案数就是n+n*(n-1)/2
//这样一定爆int,所以最好使用long long
#include<iostream>

using namespace std;

const int N=1e5+10;

int n,k;
int temp[N],total[N],cnt[N];
long long t;

int main()
{
    scanf("%d%d",&n,&k);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&temp[i]);
        total[i]=(total[i-1]+temp[i])%k;
        cnt[total[i]]++;
    }
    for(int i=0;i<k;i++)
    {
        t+=(long long)cnt[i]*(cnt[i]-1)/2;
    }
    t+=cnt[0];
    printf("%lld",t);
    return 0;
}