Cod sursa(job #2090805)

Utilizator ruxandramateiMatei Ruxandra ruxandramatei Data 18 decembrie 2017 19:08:32
Problema Secventa 5 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.92 kb
#include <iostream>
#include <fstream>
#include <cstring>
#define DMAX 2000000

using namespace std;

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

int array[DMAX], n, l ,u ;

void citire(){
    in >> n >> l >> u;
    for(int i = 1; i <= n; i++){
        in >> array[i];
    }
}

int numarSecvente(int limita){
    int frecv[DMAX];
    int nrElemDis = 0;
    int index = 1;
    int calcul = 0;
    memset(frecv, 0, sizeof(frecv));
    for(int i = 1; i <= n; i++){
        if(frecv[array[i]] == 0){
            nrElemDis++;
        }
        frecv[array[i]]++;
        while(nrElemDis > limita){
            frecv[array[index]]--;
            if(frecv[array[index]] == 0){
                nrElemDis --;
            }
            index++;
        }
        calcul += i - index;
    }
    return calcul;
}

void rezolvare(){
    out << numarSecvente(u) - numarSecvente(l - 1);
}

int main() {
    citire();
    rezolvare();
    return 0;
}