Cod sursa(job #3128464)

Utilizator alexcostinCostin Alexandru alexcostin Data 9 mai 2023 16:49:47
Problema Secventa 5 Scor 100
Compilator cpp-64 Status done
Runda Teme Pregatire ACM Unibuc 2014, Anul I Marime 1.2 kb
#include <bits/stdc++.h>
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;
}

long long calcul (int dif) {
    long long nrSec = 0;
    int nrDif = 0, p = 1;
    for (int i = 1; i <= N; i++) {
        F[i] = 0;
    }
    for (int 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;
}

void normalizare() {
    sort(v + 1, v + N + 1, 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;
    }
}

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;
}