Cod sursa(job #1446201)

Utilizator stropitoare98stropitoare stropitoare98 Data 31 mai 2015 23:53:54
Problema Divk Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.44 kb
#include <fstream>
using namespace std;
ifstream fin("divk.in");
ofstream fout("divk.out");
int v[500001],cnt[100001],n,k,a,b,x,i;
long long s;
int main()
{
    fin>>n>>k>>a>>b;
    for(i=1;i<=n;i++)
    {
        fin>>x;
        v[i]=(v[i-1]+x)%k;
    }
    cnt[0]++;
    for(i=a;i<=n;i++)
    {
        if(i-b>0)
            cnt[v[i-b-1]]--;
        s+=cnt[v[i]];
        cnt[v[i-a+1]]++;
    }
    fout<<1ll*s;
    return 0;
}