Cod sursa(job #1342403)

Utilizator badea_adi1999Badea Adrian Catalin badea_adi1999 Data 13 februarie 2015 23:13:16
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.47 kb
#include <fstream>
#define nmax 500005
#define kmax 100005
using namespace std;
ifstream f("divk.in");
ofstream g("divk.out");
long long sol;
int n,k,a,b;
int v[nmax];
int p[kmax];

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

        sol+=1LL*p[v[i]];
    }

    g<<sol;

    return 0;
}