K倍区间-前缀和进阶

给定一个长度为 N 的数列，$A_1,A_2,\dots A_N$，如果其中一段连续的子序列 $A_i,A_{i+1},\dots A_j$ 之和是 K 的倍数，我们就称这个区间 [i,j] 是 K 倍区间。

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

输入格式
第一行包含两个整数 N 和 K。

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

输出格式
输出一个整数，代表 K 倍区间的数目。

数据范围
1≤N,K≤100000,
1≤$A_i$≤100000
输入样例：

5 2
1
2
3
4
5

输出样例：

6

优化思路

1. 最先想到的是暴力求解：两重循环，遍历所有的组合情况，再区间求和。此时的时间复杂度$O(n^3)$远大于$10^8$
2. 下一步优化：区间求和可以用前缀和来优化：全局[L, R]上的和为sum[R]-sum[L-1]，此时算法优化到了$O(n^2)$
3. 再进一步优化：我们发现第二层循环的功能就是统计在$j$之前，有多少个符合条件的位置$i$，使得区间[i, j]上的和是k的倍数，即可得以下关系：
   $$\begin{gathered} (s[j]-s[i-1])modk=0 \\ s[j]modk=s[i-1]modk \end{gathered}$$
   即就是求有多少对$s[]$除以k的余数是相同的
   放在整体来看，答案就是在[1,n]的区间上，有多少对$s[]$除以k的余数相同，注意这里当i=1时，会有$s[0]$的出现，$s[0]$的值就是0（可以理解是数列中单独一个数求和的情况），这是一开始就存在的，所以在计数之前，要初始化cnt[0]=1（cnt[x]时指余数为x有多少对）（不理解的话请看完代码后看下面的解释）

代码

#include <iostream>
using namespace std;

const int N = 1e5+5;
int sum[N];
long long cnt[N];
int n,k,a;
long long res;

int main()
{
    cin>>n>>k;
    cnt[0]=1;
    for(int i=1; i<=n; i++)
    {
        cin>>a;
        sum[i] = (sum[i-1]+a)%k;
        res += cnt[sum[i]];
        cnt[sum[i]]++;
    }      
    cout<<res;
}

		res += cnt[sum[i]];
        cnt[sum[i]]++;

这两句代码其实就是数学公式$C_n^2$的过程：
$C_n^2=\frac{n\times (n-1)}{2}=1+2+3+...+n-1$
比如在求余数为1的对数时，在i=1时余数为1，此时res+=cnt[1]=0，cnt[1]++，在i=3的时候发现余数也为1，此时sum[3-1-1]满足k倍区间，所以res+=cnt[1]=1,，cnt[1]++；在i=5时发现余数也为1，此时是不是sum[5-3-1]满足k倍区间，sum[5-1-1]也满足k倍区间(两两组合)，所以res+=cnt[1]=1+2=3…依次类推整个过程就是高中所学的$C_n^2$

如果你不理解为什么cnt[0]=1，可以按照上述流程模拟余数为0的时候，可以发现余数不为0的时候，需要最少有两个这种区间才能组合一起计一个k倍区间，而余数为0的，其自己本身就是k倍区间，和别人组合也是k倍区间，所以一开始要初始化，不然就少了组合
如果还不理解，也可以不初始化，在最后给res加上所有单独一个数就满足k倍的个数

代码

#include <iostream>
using namespace std;

const int N = 1e5+5;
int sum[N];
long long cnt[N];
int n,k,a;
long long res;

int main()
{
    cin>>n>>k;
    cnt[0]=1;
    for(int i=1; i<=n; i++)
    {
        cin>>a;
        sum[i] = (sum[i-1]+a)%k;
        cnt[sum[i]]++;
    }
    
    for(int i=0; i<k; i++)
        res += (cnt[i]*(cnt[i]-1)/2);
    cout<<res;
}