Cod sursa(job #1866097)

Utilizator KronSabau Valeriu Kron Data 2 februarie 2017 17:29:08
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.49 kb
#include <iostream>
#include <fstream>
using namespace std;
int n,a,b,k;
int sum[500015],pos[100015];
ifstream f("divk.in");
ofstream g("divk.out");

int main()
{
    int x;
    f >> n >> k >> a >> b;
    for(int i=1;i<=n;i++)
    {
        f >> x;
        sum[i]=(sum[i-1]+x)%k;
    }
    long long  ras=0;
    for(int i=a;i<=n;i++)
    {
        pos[sum[i-a]]++;
        if(i>b)
            pos[sum[i-b-1]]--;
        ras+=pos[sum[i]];
    }
    g << ras;
    return 0;
}