Cod sursa(job #2467079)

Utilizator Simon2712Simon Slanina Simon2712 Data 3 octombrie 2019 17:49:38
Problema Divk Scor 20
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.5 kb
#include <fstream>

using namespace std;
ifstream cin("divk.in");
ofstream cout("divk.out");
int sum[500001];
int vc[100000],vca[100001];
int main()
{
    int n,a,b,rez=0,k,nr,i;
    long long s=0;
    cin>>n>>k>>a>>b;
    for(i=1;i<=n;i++)
    {
        cin>>nr;
        s+=nr;
        sum[i]=s%k;
        if(i>b)
            vc[sum[i-b]]--;
        vca[s%k]++;
        if(i>=a){
          vca[sum[i-a+1]]--;
          vc[sum[i-a+1]]++;
        }
        rez+=vc[s%k];
    }
    cout<<rez;
    return 0;
}