Cod sursa(job #2462743)

Utilizator Alex_DiaconuDiaconu Alexandru Alex_Diaconu Data 27 septembrie 2019 19:35:27
Problema Divk Scor 90
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.54 kb
#include <fstream>

using namespace std;

ifstream ci ("divk.in");
ofstream co ("divk.out");

long long nr[100001];

int main()
{
    int n,a,b,k;
    ci >> n >> k >> a >> b;
    long long v[n+1];
    v[0]=0;
    for (int i=1; i<=n; i++)
    {
        ci >> v[i];
        v[i]=v[i]+v[i-1];
        v[i]=v[i]%k;
    }
    int r=0;
    for (int i=a; i<=b; i++)
    {
        nr[v[i-a]]++;
        r+=nr[v[i]];
    }
    for (int i=b+1; i<=n; i++)
    {
        nr[v[i-a]]++;
        nr[v[i-b-1]]--;
        r+=nr[v[i]];
    }
    co << r;
    return 0;
}