Cod sursa(job #2155533)

Utilizator Constantin.Dragancea Constantin Constantin. Data 7 martie 2018 21:58:01
Problema Divk Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.81 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 (st < dr && v[i][dr] - v[i][st] > l) st++;
                while (dr < v[i].size() && v[i][dr] - v[i][st] <= l) dr++;
                ans += dr - st - 1;
                st++;
            }
        }
    }
    return ans;
}

int main(){
    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])%k, v[dp[i]].push_back(i);
    cout << get(b) - get(a - 1);
    return 0;
}