Cod sursa(job #2156042)

Utilizator Constantin.Dragancea Constantin Constantin. Data 8 martie 2018 13:53:41
Problema Divk Scor 10
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.89 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 (dr + 1 < v[i].size() && v[i][dr + 1] - v[i][st] <= l) dr++;
            while (dr < v[i].size()){
                while (st < dr && v[i][dr] - v[i][st] > l) st++;
                if (dr == v[i].size() - 1) ans += (dr - st)*(dr - st + 1)/2;
                else ans += dr - st;
                dr++;
            }
        }
    }
    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]), v[dp[i]%k].push_back(i);
    cout << get(b) - get(a - 1);
    return 0;
}