Cod sursa(job #1233780)

Utilizator japjappedulapPotra Vlad japjappedulap Data 26 septembrie 2014 00:51:52
Problema Divk Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.49 kb
#include <fstream>
using namespace std;

ifstream is ("divk.in");
ofstream os ("divk.out");

int N, A, B, K;
int v[500003], r[100003];
int S;

int main()
{
    is >> N >> K >> A >> B;
    for (int i = 1, x; i <= N; ++i)
        is >> x, v[i] = (v[i-1] + x) % K;
    for (int i = 0; i <= N; ++i)
    {
        if (i >= A)
            r[v[i-A]]++;
        if (i-1 >= B)
            r[v[i-B-1]]--;
        S += r[v[i]];
    }
    os << S;
    is.close();
    os.close();
}