Cod sursa(job #2863645)

Utilizator cdenisCovei Denis cdenis Data 7 martie 2022 00:51:36
Problema Secventa 5 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.83 kb
#include <iostream>
#include <fstream>
#include <unordered_map>
#include <algorithm>

using namespace std;

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

int n,u,l,nr;
unsigned int v[1<<20+1],ap[1<<20+1],f[1<<20+1];

long long secv(int k)
{
    for(int i=1;i<=nr;i++)
        ap[i]=0;
    long long st=1,cnt=0,sz=0;
    for(int dr=1;dr<=n;dr++)
    {
        if(!ap[f[v[dr]]])
            sz++;
        ap[f[v[dr]]]++;
        while(sz>k)
        {
            ap[f[v[st]]]--;
            if(ap[f[v[st]]]==0)
                sz--;
            st++;
        }
        cnt+=dr-st+1;
    }
    return cnt;
}

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