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
求区间[l,r]的和是k的倍数的个数。求区间和，我们可以通过前缀和来求出。我们规定sum[i]表示第1个元素到第i个元素的和。那么sum[r] - sum[l-1]就是区间[l,r]的和。区间[l,r]的和是k的倍数即(sum[r] - sum[l-1])%k == 0 即sum[r]%k == sum[l-1]%k

#include <cstdio>
#include <cstring>
#include <iostream>
#include <algorithm>

using namespace std;

typedef long long LL;

const int N = 100010;

int n, k;
LL s[N], cnt[N];

int main()
{
    scanf("%d%d", &n, &k);
    for (int i = 1; i <= n; i ++ )
    {
        scanf("%lld", &s[i]);
        s[i] += s[i - 1];
    }

 //   LL res = 0;
    cnt[0] = 1;// s[0]=0; 某一区间和为s[r]-s[l-1]，l=1是就出现s[0]了
    //为什么赋值为1？
//因为我们的思路是找两个序列和a%k和b%k的余数相同的个数
//而我们的前缀和一般是不包含S0这个东西的，因为没有意义，但是这道题有意义
//样例里面前缀和序列%k之后是1 1 0 0 1，两两比较，我们只能找到四个
//为什么少了两个？因为我们不一定需要两个序列，单个序列取余=0也构成K倍区间
//也就是说s3 s4这两个区间是能单独成K倍区间的，而我们的思路是找两个序列比较
//此时，我们就要假设S0=0是有意义的，我们就可以有(s0，s3),(s0,s4)这两个组合
    for (int i = 1; i <= n; i ++ )
    {
    //    res += cnt[s[i] % k]; // 当有二个数的余数相等了才加1，所以后cnt[]++;
        cnt[s[i] % k] ++ ;
    }

  //  printf("%lld\n", res);
    LL ans=0;
    for(int i=0;i<=k-1;i++){
        ans+=(cnt[i]*(cnt[i]-1))/2;    //意思是
       // 从（cnt【i】=5）即5个人里面选两个有几种选法
    }
    cout << ans;
    return 0;
}



#include<bits/stdc++.h>
#define int long long
using namespace std;
const int N=1e5+10;
int a[N];
int num[N];
signed main(){
    int n,k;
    cin >> n >> k;
    for(int i=1;i<=n;i++){
        cin >> a[i];
        a[i]+=a[i-1];
        num[a[i]%k]++;
    }
    int res=0;
    for(int i=0;i<k;i++){
        if(num[i]>1){
            res+=num[i]*(num[i]-1)/2;
        }
    }
    for(int i=1;i<=n;i++){
        if(a[i]%k==0) res++;
    }
    cout << res;
    return 0;
}