Cod sursa(job #1332725)

Utilizator vazanIonescu Victor Razvan vazan Data 2 februarie 2015 12:47:53
Problema Divk Scor 20
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.77 kb
#include<cstdio>
#include<vector>
using namespace std;
int spart=0;
vector<int> v[100000];
int main()
{
    FILE *in=fopen("divk.in", "r"),
         *out=fopen("divk.out", "w");
    int n, k, a, b, tmp, rasp=0, j, i, q;
    fscanf(in, "%d%d%d%d", &n, &k, &a, &b);
    for(i=0; i<=n; i++)
    {
        fscanf(in, "%d", &tmp);
        v[spart].push_back(i);
        spart=(spart+tmp)%k;
    }
    for(i=0; i<k; i++)
    {
        j=0; q=1;
        while(q<v[i].size())
        {
            if(v[i][q]-v[i][j]>=a && v[i][q]-v[i][j]<=b)
                rasp++;
            if(!v[i][q]-v[i][j]<a && j<q-1)
                j++;
            else
                q++;
        }
    }
    fprintf(out, "%d", rasp);
    fclose(in);
    fclose(out);
    return 0;
}