蓝桥杯 k倍区间

链接
sum[i]存储余数为i出现的次数
当为i时，sum[1]=1,为j时，sum[j]=2，则i到j区间满足题意，但是当sum[0]出现的时候，会遗漏，需要单独考虑

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
typedef long long ll;
const int maxn=1e5+10;
ll a[maxn],sum[maxn],b[maxn];
int main()
{
	int n,k;
	ll ans=0;
	scanf("%d%d",&n,&k);
	memset(sum,0,sizeof sum);
	for(int i=1;i<=n;i++)
	{
		scanf("%lld",&a[i]);
		sum[i]=(sum[i-1]+a[i])%k;
		ans+=b[sum[i]];
		b[sum[i]]++;
	}
	printf("%lld\n",ans+b[0]);
	return 0;
}