Cod sursa(job #2181850)

Utilizator grecubogdanGrecu Bogdan grecubogdan Data 21 martie 2018 21:20:36
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.4 kb
#include <fstream>
using namespace std;
ifstream f("divk.in");
ofstream g("divk.out");
int n,k,a,b,i,j,nr;
long long x[500010],rest[100002],rz;
int main()
{
    f>>n>>k>>a>>b;
    for(i=1;i<=n;i++)
    {
        f>>x[i];
        x[i]=(x[i]+x[i-1])%k;
        if(i>=a)   rest[x[i-a]]++;
        if(i-b>=1) rest[x[i-b-1]]--;
        rz+=rest[x[i]];
    }
    g<<rz<<'\n';
    return 0;
}