前缀和求k倍区间

蓝桥杯（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

思路：

用前缀和求出从一到最后各段的和，然后如果前面有一段和后面的某段对k取模是相等的，那么说明后一段减去前一段是k的整数倍（余数一样相剪余数不就等于0了吗）这里赋一个大佬讲解的前缀和，我看了茅塞顿开：http://t.csdn.cn/EqbtL

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
long long a[N];
int main()
{
    int n,k,i;
    long long s=0;
    scanf("%d%d",&n,&k);
    int count[n]={0};
    for(i=1;i<=n;i++)
    {scanf("%lld",&a[i]);
     a[i]+=a[i-1];
    }
    count[0]=1;
    for(i=1;i<=n;i++)
    {s+=count[a[i]%k];
     count[a[i]%k]++;
    }
    printf("%lld",s);
    return 0;
}

题外

这道题是我学了前缀和找的题，换做以前我会傻嗨嗨的用暴力解法一遍遍历🥹 活该我以前超时，后来学了就尽量优化点算法