Cod sursa(job #2238130)

Utilizator GiihuoTihufiNeacsu Stefan GiihuoTihufi Data 4 septembrie 2018 18:01:12
Problema Divk Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.47 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("divk.in");
ofstream g("divk.out");

int N,K,A,B;
int S[500001],V[100001];

int main()
{
    f>>N>>K>>A>>B;
    for(int i=1,x;i<=N;i++)
    {
        f>>x;
        S[i]=(x+S[i-1])%K;
    }
    uint64_t result=0;
    for(int i=1;i<A;i++) V[S[i]]++;
    for(int i=A;i<N;i++)
    {
        V[S[i-B+1]]--;
        V[S[i-A+2]]++;
        result+=V[S[i]];
    }
    g<<result;

    return 0;
}