Cod sursa(job #3360080)

Utilizator AkysAkys Miron Akys Data 8 iulie 2026 16:19:29
Problema Divk Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.55 kb
#include <fstream>

using namespace std;
ifstream fin("divk.in");
ofstream fout("divk.out");
int n,k,a,b,A[500001],F[100001]={};
long long S[500001]={};

int main()
{
    int c=0;
    fin>>n>>k>>a>>b;
    for(int i=1;i<=n;i++)
    {
        fin>>A[i];
        S[i]=S[i-1]+A[i];
    }
    for(int i=1;i<=n;i++)
        {
            S[i]=S[i]%k;
        }

    for(int i=1;i<=n;i++)
    {
        if(i>=a)
            F[S[i-a]]++;
        if(i>b)
            F[S[i-b-1]]--;
        c=c+F[S[i]];
    }
    fout<<c;
    return 0;
}