Cod sursa(job #2095487)

Utilizator zhm248Mustatea Radu zhm248 Data 27 decembrie 2017 16:32:33
Problema Divk Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.64 kb
#include<cstdio>
using namespace std;
int s[500001],nr[100001];
int main()
{
    freopen("divk.in","r",stdin);
    freopen("divk.out","w",stdout);
    int n,k,a,b,i,x,poz=0;
    long long sol=0;
    scanf("%d%d%d%d",&n,&k,&a,&b);
    for(i=1;i<=n;++i)
    {
        scanf("%d",&x);
        s[i]=(s[i-1]+x)%k;
    }
    for(i=a;i<=n;++i)
    {
        nr[s[i]]++;
        if(i==b+poz)
        {
            sol+=nr[s[poz]];
            nr[s[a+poz]]--;
            poz++;
        }
    }
    while(n>=a+poz)
    {
        sol+=nr[s[poz]];
        nr[s[a+poz]]--;
        poz++;
    }
    printf("%d\n",sol);
    return 0;
}