【蓝桥杯C++】k倍区间

本题为，巧用前缀和+数学表达式转换

优化为O(N)，

代码思路

#include <iostream>
#include <cstdio>
using namespace std;

const int N = 100010;
typedef long long ll;

ll s[N] ;
int cnt[N];

int n , k;

int main()
{
    scanf("%d%d" , &n , &k );
    for( int i = 1 ; i <= n ; i++ ) 
    {
        scanf("%d" , &s[i]); 
        s[i] += s[i-1];
    }
    
    ll res = 0;
    cnt[0] = 1;//cnt[0]中存的是s[]中等于0的数的个数 由于s[0] = 0 所以最初等于0的有1个数 所以cnt[0] = 1 
    for( int i = 1 ; i <= n ; i++ )
    {
        res += cnt[s[i] % k];//表示s[i]%k的余数出现的次数
        cnt[s[i] % k]++;
    }
    printf("%lld",res);
    return 0;
}

要注意的就是，int会爆栈，开long long