Cod sursa(job #3298357)

Utilizator Octavian09Dore Octaviam Octavian09 Data 29 mai 2025 08:06:02
Problema Secventa 5 Scor 100
Compilator cpp-64 Status done
Runda Teme Pregatire ACM Unibuc 2014, Anul I Marime 1.08 kb
#include <iostream>
#include <algorithm>
#include <fstream>

using namespace std;
const int NMAX=(1<<20)+1;

struct Element{
    unsigned int val;
    int poz;
};

Element v[NMAX];

int N, w[NMAX], F[NMAX];

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

bool comp(const Element& x,const Element& y){
    return x.val < y.val;
}

void normalizare(){
    sort(v+1,v+1+N,comp);
    int x=1;
    w[v[1].poz]=1;
    for(int i=2;i<=N;i++){
        if(v[i].val != v[i-1].val) ++x;
        w[v[i].poz]=x;
    }
}

long long calcul(int dif){
    long long nrSec=0;
    int nrDif=0,p=1,i;
    for(i=1;i<=N;i++) F[i]=0;
    for(i=1;i<=N;i++){
        if(++F[w[i]]==1) ++nrDif;
        while(nrDif> dif && p <= i){
            if(--F[w[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].val;
        v[i].poz=i;
    }
    normalizare();
    g << calcul(U)-calcul(L-1);
    f.close();
    g.close();
    return 0;
}