Cod sursa(job #1886511)

Utilizator adystar00Bunea Andrei adystar00 Data 20 februarie 2017 22:24:28
Problema Divk Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.5 kb
#include <iostream>
#include <fstream>
using namespace std;
int v[500010],vc[100010];
int main()
{
    ifstream fin ("divk.in");
    ofstream fout ("divk.out");
    int n,k,a,b,i,suma=0,x;
    fin>>n>>k>>a>>b;
    for(i=1; i<=n; i++)
    {
        fin>>x;
        v[i]=v[i-1]+x;
        v[i]%=k;
    }
    for(i=2; i<=n; i++)
    {
        if(i-b>0)
            vc[v[i-b]]--;
        if(i-a>=0)
            vc[v[i-a+1]]++;
        suma=suma+vc[v[i]];
    }
    fout<<suma;
    return 0;
}