Cod sursa(job #3128463)

Utilizator oana75Ioana Prunaru oana75 Data 9 mai 2023 16:47:58
Problema Secventa 5 Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.23 kb
#include <fstream>
#include <algorithm>

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 in("secv5.in");
ofstream out("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(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;
}

int main()
{
    int L, U;
    in >> N >> L >> U;
    for(int i = 1; i <= N; i++)
    {
        in >> v[i].val;
        v[i].poz = i;
    }
    normalizare();
    out << calcul(U) - calcul(L - 1);
    in.close();
    out.close();
    return 0;
}