蓝桥杯之k倍区间

给定一个长度为N的数列，A1, A2, ... AN，如果其中一段连续的子序列Ai, Ai+1, ... Aj(i <= j)之和是K的倍数，我们就称这个区间[i, j]是K倍区间。 你能求出数列中总共有多少个K倍区间吗？ 输入-----第一行包含两个整数N和K。(1 <= N, K <= 100000) 以下N行每行包含一个整数Ai。(1 <= Ai <= 100000) 输出-----输出一个整数，代表K倍区间的数目。 例如，

输入：5 2 

1 

2 

3 

4 

5 

程序应该输出：6

首先统计前缀和sum[i] 表示A1+A2+…+Ai.所以对于任意一段区间[l,r]的和就是sum[r]-sum[l-1].

如果要保证这个区间和为K倍数就是：(sum[r]-sum[l-1])%k == 0.变形后就是：sum[r]%k==sum[l-1]%k，所以我们计算前缀和的时候顺带模K

#include <algorithm>
#include <string.h>
#include <iostream>
#include <stdio.h>
#include <string>
#include <vector>
#include <queue>
#include <map>
#include <set>
using namespace std;
typedef long long ll;
ll bk[100010]= {0};
ll arr[100010];
ll k,n;
int main()
{
    scanf("%lld%lld",&n,&k);
    for(int i = 0 ; i < n ; i ++)
        scanf("%lld",&arr[i]);
    arr[0] %= k;
    ll sum = 0;
    for(int i = 1 ; i < n ; i ++)
    {
        arr[i] = (arr[i]+arr[i-1])%k;
    }
    for(int i = 0 ; i < n ; i ++)
    {
        sum =sum+bk[arr[i]];//不是很理解这里，可以简单的理解为中间过程相加余k等于0的个数
        bk[arr[i]]++;
    }
    printf("%lld\n",sum+bk[0]);//bk[0]就是从arr[0]加到某个数的和 余k等于0（也就是刚好能整除k），所以有几个，就加上几个；
    return 0;
}

就拿这个测试来说，经过第一个for循环后 arr数组最后变成了 1  1  0  0  1;

第二个for循环中的bk数组可以理解为sum加上arr[i]曾出现 过  得次数，注意：我说的是已经出现过了的次数，不包括现在出现的这次;

比如 第一个1的时候1出现过了0次，而不是1次；第二个1的时候1出现了1次，而不是2次；第一个0出现了0次，以此类推