Cod sursa(job #1142072)

Utilizator andrei_diaconuAndrei Diaconu andrei_diaconu Data 13 martie 2014 14:46:47
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.42 kb
#include <fstream>
#define NMax 500001
using namespace std;
ifstream f("divk.in");
ofstream g("divk.out");
long long n, k, a, b, v[NMax], r[NMax], i, s, el;
int main()
{
    f>>n>>k>>a>>b;
    for (i=1; i<=n; i++) {
        f>>el;
        v[i]=(v[i-1]+el)%k;
    }
    for (i=a; i<=n; i++) {
        r[v[i-a]]++;
        if(i>b)
            r[v[i-b-1]]--;
        s+=r[v[i]];
    }
    g<<s;
    return 0;
}