Cod sursa(job #2156117)

Utilizator Constantin.Dragancea Constantin Constantin. Data 8 martie 2018 14:46:13
Problema Divk Scor 30
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.8 kb
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int n, k, a, b, dp[500010];
vector <int> v[100005];

ll get(int l){
    ll ans = 0;
    for (int i=0; i<k; i++){
        if (v[i].size() > 1){
            int st = 0, dr = 0;
            while (st < v[i].size()){
                while (dr + 1 < v[i].size() && v[i][dr + 1] - v[i][st] <= l) dr++;
                ans += dr - st;
                st++;
            }
        }
    }
    return ans;
}

int main(){
    ios_base::sync_with_stdio(0); cin.tie(0);
    ifstream cin ("divk.in");
    ofstream cout ("divk.out");
    cin >> n >> k >> a >> b;
    v[0].push_back(0);
    for (int i=1; i<=n; i++) cin >> dp[i], dp[i] = (dp[i] + dp[i-1]), v[dp[i]%k].push_back(i);
    cout << get(b) - get(a - 1);
    return 0;
}