Cod sursa(job #2508894)

Utilizator cyg_SerbanBFlorin Gheorghe cyg_SerbanB Data 13 decembrie 2019 12:43:31
Problema Divk Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.47 kb
#include <cstdio>
using namespace std;
int sp[500005],cnt[100005];
int main()
{
    freopen("divk.in","r",stdin);
    freopen("divk.out","w",stdout);
    int n,i,x,k,a,b;
    long long ans=0;
    scanf("%d%d%d%d",&n,&k,&a,&b);
    for(i=1;i<=n;++i)
    {
        scanf("%d",&x);
        sp[i]=(sp[i-1]+x)%k;
        if(i>b)
            cnt[sp[i-b-1]]--;
        if(i>=a)
            cnt[sp[i-a]]++;
        ans=ans+1LL*cnt[sp[i]];
    }
    printf("%lld",ans);
    return 0;
}