Cod sursa(job #2863607)

Utilizator cdenisCovei Denis cdenis Data 6 martie 2022 23:56:07
Problema Secventa 5 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <iostream>
#include <fstream>
#include <unordered_map>

using namespace std;

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

const int MAX=1<<20+1;
int n,v[MAX],i,st,dr,u,l;
long long cnt;
unordered_map < unsigned int , int > ap;

long long secv(int k)
{
    ap.clear();
    st=1;
    cnt=0;
    for(dr=1;dr<=n;dr++)
    {
        ap[v[dr]]++;
        while(ap.size()>k)
        {
            ap[v[st]]--;
            if(!ap[v[st]])
                ap.erase(v[st]);
            st++;
        }
        cnt+=dr-st+1LL;
    }
    return cnt;
}

int main()
{
    fin >> n >> l >> u;
    for(i=1;i<=n;i++)
        fin >> v[i];
    fout << secv(u)-secv(l-1);
	return 0;
}