Cod sursa(job #2868647)

Utilizator PalcPalcu Stefan Palc Data 11 martie 2022 08:49:15
Problema Secventa 5 Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.79 kb
#include <bits/stdc++.h>
using namespace std;

ifstream fin("secv5.in");
ofstream fout("secv5.out");

long long a[1050005];
int n, l, u;
unordered_map<long long, int> M;

long long Secv(int x)
{
    int i, nrdist = 0, p = 1;
    long long cnt = 0;
    for (i = 1; i <= n; i++)
    {
        M[a[i]]++;
        if (M[a[i]] == 1) nrdist++;
        while (nrdist > x)
        {
            M[a[p]]--;
            if (M[a[p]] == 0) nrdist--;
            p++;
        }
        cnt += (i - p + 1);
    }
    return cnt;
}
int main()
{
    int i;
    long long x, y;
    fin >> n >> l >> u;
    for (i = 1; i <= n; i++)
        fin >> a[i];
    x = Secv(u);
    M.clear();
    y = Secv(l - 1);
    fout << x - y << "\n";
    fin.close();
    fout.close();
    return 0;
}