Cod sursa(job #3207456)

Utilizator raresmihai1234Rares Mihai raresmihai1234 Data 26 februarie 2024 10:49:03
Problema Divk Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.51 kb
#include <fstream>

using namespace std;
ifstream fin("divk.in");
ofstream fout("divk.out");
int a,b,i,s[500005],inc,sfar,nrh,suma,n,k,x,j,sum,vf[500005],nr;
int main()
{
    fin>>n>>k>>a>>b;
    for (i=1; i<=n; i++)
    {
        fin>>x;
        s[i]=(s[i-1]+x)%k;
    }
    sum=0;
    if (b<a) swap(a,b);
    for (j=1;j<=n;j++)
    {
        if (j>=a)
        {
           vf[s[j-a]]++;
        }
        nr+=vf[s[j]];
        if (j>=b) vf[s[j-b]]--;
    }
    fout<<nr;


    return 0;
}