Cod sursa(job #3185796)

Utilizator dianaionicaDiana Ionica dianaionica Data 20 decembrie 2023 14:16:07
Problema Secventa 5 Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.74 kb
#include <fstream>
#include <unordered_map>
using namespace std;
ifstream cin("secv5.in");
ofstream cout("secv5.out");
int n;
long long int v[1<<20];

int rasp(int x)
{
    unordered_map < long long int, int > f;
    long long int cnt = 0, ans;
    int i, j;
    i = j = ans = 0;
    while(i < n)
    {
        f[v[i]]++;
        if(f[v[i]] == 1)
            cnt++;
        i++;

        while(cnt > x)
        {
            f[v[j]]--;
            if(f[v[j]] == 0)
                cnt--;
            j++;
        }
        ans += i - j;
    }
    return ans;
}
int main()
{
    int a, b;
    cin >> n >> a >> b;
    for(int i = 0; i < n; i++)
        cin >> v[i];
    cout << rasp(b) - rasp(a - 1);
    return 0;
}