Cod sursa(job #3274015)

Utilizator drsbosDarius Scripcaru drsbos Data 4 februarie 2025 19:28:39
Problema Secventa 5 Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.06 kb
#include <fstream>
#include <stack>
#include <queue>
#include <cmath>
#include <algorithm>
#include <iostream>
#include <set>
#include <cstring>
#include <map>
#include <string>
#include<iomanip>
#include<bitset>
#include<unordered_map>
#define oo 2000000000
#define MOD 777013
using namespace std;

ifstream fin("secv5.in");
ofstream fout("secv5.out");
int l, u, n, k, rez;
unsigned int a[1100005];
unordered_map<unsigned long long, int>m;
unsigned long long nrsecv(int k)
{
    int st = 1, dr = 1, cnt = 0;
    unsigned long long rez = 0;
    m.clear();
    while (dr <= n)
    {
        if (m[a[dr]] == 0)
            cnt++;
        m[a[dr]]++;
        while (cnt > k)
        {
            m[a[st]]--;
            if (m[a[st]] == 0)
                cnt--;
            st++;
        }
        rez += (dr - st + 1);
        dr++;
    }
    return rez;
}
int main()
{
    fin >> n >> l >> u;
    for (int i = 1; i <= n; i++)
    {
        fin >> a[i];
    }
    fout<<nrsecv(u)-nrsecv(l-1);
}