Cod sursa(job #2697514)

Utilizator Diana_IonitaIonita Diana Diana_Ionita Data 18 ianuarie 2021 20:08:22
Problema Divk Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.49 kb
#include <bits/stdc++.h>
using namespace std;
ifstream fin("divk.in");
ofstream fout("divk.out");
int n,k,aa,b,i,x,rez,a[500001],dp[500001];
int main()
{
    fin>>n>>k>>aa>>b;
    for (i=1; i<=n; i++)
    {
        fin >>x;
        a[i]=a[i-1];
        a[i]+=x;
        a[i]%=k;
    }
    for (i=1; i<=n; i++)
    {
        if(i>=aa)
            dp[a[i-aa]]++;
        if(i>b)
            dp[a[i-b-1]]--;
        rez+=dp[a[i]];
    }

    fout<<rez<< '\n';
    return 0;
}