Cod sursa(job #3151928)

Utilizator KaleemRazaSyedKaleem Raza Syed KaleemRazaSyed Data 23 septembrie 2023 09:39:05
Problema Divk Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.55 kb
#include<iostream>

using namespace std;

const int N = 10000005;
int pref[N];
int n, k;

long long find_ans(int b)
{
  int cnt[k] = {0};
  long long ans = 0;

  for(int i = 0; i < b; ++i)
    ans += cnt[pref[i]]++;

  for(int i = b; i <= n; i ++)
    cnt[pref[i-b]]-- , ans += cnt[pref[i]]++;
  return ans;
}

int main()
{
  int s, e;
  cin >> n >> k >> s >> e;
  for(int i = 1; i <= n; i ++)
    cin >> pref[i];

  pref[0] = 0;
  for(int i = 1; i <=n ; i++)
    pref[i] = (pref[i] + pref[i-1]) % k;
  
  cout << find_ans(e+1) - find_ans(s) << "\n";

  return 0;
}