Cod sursa(job #12848)

Utilizator victorsbVictor Rusu victorsb Data 4 februarie 2007 22:33:00
Problema Secventa 5 Scor 90
Compilator c Status done
Runda Arhiva de probleme Marime 1.79 kb
#include <stdio.h>

const int Nmax = 1048576;
const unsigned int r1 = 2654435761u;
const unsigned int r2 = 1666666013u;
const int Tmax1 = 454321;
const int Tmax2 = 196613;
int i, j, n, l, u, ct, h1, h2, st, dr, sz1, sz2;
long long sol;
unsigned int sir[1048576], vl;
int t1[454321][5], t2[196613][5];
int s[1048576], v[1048576];

inline void citire()
{
    scanf("%d %d %d\n", &n, &l, &u);
    for (i=1; i<=n; ++i)
        scanf("%u", &sir[i]);
}

inline long long solve2(int m)
{
    long long ret = 0;
    st = 1;
    ct = 0;
    for (dr=1; dr<=n; ++dr)
    {
        if (!v[s[dr]])
            ++ct;
        ++v[s[dr]];
        while (ct > m)
        {
            if (!(--v[s[st]]))
                --ct;
            ++st;
        }
        ret += dr - st + 1;
    }
    for (; st<=n; ++st)
        --v[s[st]];
    return ret;
}

inline int find(int x)
{
    int i;
    for (i=1; i<=sz1; ++i)
        if (vl == sir[t1[h1][i]])
        {
            s[x] = s[t1[h1][i]];
            return 1;
        }
    for (i=1; i<=sz2; ++i)
        if (vl == sir[t2[h2][i]])
        {
            s[x] = s[t2[h2][i]];
            return 1;
        }
    return 0;
}

inline void solve()
{
    for (i=1; i<=n; ++i)
    {
        vl = sir[i];
        h1 = (vl * r1) % Tmax1;
        h2 = (vl * r2) % Tmax2;
        sz1 = t1[h1][0];
        sz2 = t2[h2][0];
        if (!find(i))
        {
            s[i] = ++ct;
            if (sz1 < sz2)
                t1[h1][++t1[h1][0]] = i;
            else
                t2[h2][++t2[h2][0]] = i;
        }
    }
    sol = solve2(u) - solve2(l-1);
    printf("%lld\n", sol);
}

int main()
{
    freopen("secv5.in", "r", stdin);
    freopen("secv5.out", "w", stdout);
    citire();
    solve();
    return 0;
}