Cod sursa(job #1186338)

Utilizator andi23roscaRosca Andrei andi23rosca Data 17 mai 2014 12:48:50
Problema Divk Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.46 kb
#include <iostream>
#include <fstream>
#define maxn 500010

using namespace std;

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

long long n,k,a,b,x,i,v[maxn],r[maxn],s=0;

int main()
{
    f>>n>>k>>a>>b;
    for(i=1;i<=n;i++)
    {
        f>>x;
        v[i] = (v[i-1] + x) % k;
    }
    for(i=1;i<=n;i++)
    {
        if(i>=a)
            r[v[i-a]]++;
        if(i>b)
            r[v[i-a-1]]--;
        s+=r[v[i]];
    }
    g<<s;
}