Cod sursa(job #2038990)

Utilizator Fanika123Tanasa Stefan Fanika123 Data 14 octombrie 2017 10:30:23
Problema Secventa 5 Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.91 kb
#include <bits/stdc++.h>
#define in "secv5.in"
#define out "secv5.out"
using namespace std;
ifstream fin(in);
ofstream fout(out);

unsigned a[1024*1024 + 20];
int n , L, U;

///functie care determina nr secv cu cel mult x elemente distincte
long long NrSecv(unsigned x)
{
    unordered_map <unsigned ,int> M;
    long long cnt = 0,i;
    int j;

    i = 1;
    for (j = 1; j <= n; j++)
    {
        M[a[j]]++;
        while (M.size() > x)
        {
            M[a[i]]--;
            if (M[a[i]] == 0) M.erase(a[i]);
            i++;
        }
        cnt += (j-i+1); /// adaug cele j-i+1 secvente care se termina cu a[j]
    }
    return cnt;
}

void Citire()
{
    int i;

    fin >> n >> L >> U;
    for (i = 1; i <= n; i++)
        fin >> a[i];

    fout << NrSecv(U) - NrSecv(L-1)<< "\n";
}

int main()
{
    Citire();

    fin.close();
    fout.close();
    return 0;
}