Cod sursa(job #1145568)

Utilizator gapdanPopescu George gapdan Data 18 martie 2014 12:01:26
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.52 kb
#include<cstdio>
using namespace std;
long long n,k,a,b,i,s;
long long x[500001],rest[100001];
int main()
{
    freopen("divk.in","r",stdin);
    freopen("divk.out","w",stdout);
    scanf("%lld%lld%lld%lld",&n,&k,&a,&b);
    for (i=1;i<=n;++i)
    {
        scanf("%lld",&x[i]);
        x[i]=(x[i-1]+x[i])%k;
        if (i>=a)
        {
            ++rest[x[i-a]];
        }
        if (i>b)
        {
            --rest[x[i-b-1]];
        }
        s+=rest[x[i]];
    }
    printf("%lld\n",s);
    return 0;
}