Cod sursa(job #3122811)

Utilizator AffectiveSmile2Mihnea Matea AffectiveSmile2 Data 20 aprilie 2023 18:42:09
Problema Secventa 5 Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.72 kb
#include <iostream>
#include <fstream>
#include <unordered_map>
using namespace std;
ifstream f("secv5.in");
ofstream g("secv5.out");
const int NMAX=(1<<20+1);
int N;
unsigned v[NMAX];
long long calcul(int dif)
{
    unordered_map <unsigned,int> M;
    long long nrSec=0;
    int nrDif=0,p=1;
    for(int i=1;i<=N;i++)
    {
        if(++M[v[i]]==1)
            ++nrDif;
        while(nrDif>dif&&p<=i)
        {
            if(--M[v[p]]==0)
                --nrDif;
            ++p;
        }
        nrSec+=i-p+1;
    }
    return nrSec;
}
int main()
{
    int L,U;
    f>>N>>L>>U;
    for(int i=1;i<=N;i++)
        f>>v[i];
    g<<calcul(U)-calcul(L-1);
    f.close();
    g.close();
    return 0;
}