ACWing:1230.K倍区间

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

const int N = 1e5 + 10;

long long arr[N] = {0};

int remain[N] = {0};

int n,k;
int main(){
    ios::sync_with_stdio(0),cin.tie(0);
    cin >> n >> k;
    for(int i = 1;i <= n;i++){
       cin >> arr[i];
       arr[i] += arr[i-1];          // 计算前缀和
    }
    long long  ans=0;
    
    remain[0] = 1;          // 一开始前缀和1 ~ i 如果刚好可以整除k的话需要加一
    
    for(int i = 1;i <= n;i++){
        
        ans += remain[arr[i] % k];
        
        remain[arr[i] % k] ++;
    }
    cout << ans << endl;
    return 0;
}