Cod sursa(job #3202797)

Utilizator Raissa14Raissa Sandor Raissa14 Data 12 februarie 2024 11:39:32
Problema Divk Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.47 kb
#include <fstream>

using namespace std;
ifstream in("divk.in");
ofstream out("divk.out");
int n,k,a,b,s[500001],i,m,fr[100001];
long long t;
int main()
{
    in>>n>>k>>a>>b;
    for(i=1;i<=n;i++)
    {
        in>>m;
        s[i]=(s[i-1]+m)%k;

    }
    for(i=1;i<=n;i++)
    {
        if(i>=a)
        {
            fr[s[i-a]]++;

        } t+=fr[s[i]];
        if(i>=b)
        {
            fr[s[i-b]]--;
        }
    }
    out<<t;
    return 0;
}