Cod sursa(job #9466)

Utilizator svalentinValentin Stanciu svalentin Data 27 ianuarie 2007 15:39:19
Problema Secventa 5 Scor 80
Compilator cpp Status done
Runda Unirea 2007, clasele 11-12 Marime 1.46 kb
#include<stdio.h>
#include<string.h>

#include<algorithm>
using namespace std;

const int MAXN = 1<<20;

unsigned int nr[MAXN], nr2[MAXN];
int c1[MAXN], c2[MAXN];
int n, l, r;

int main(void)
{
    freopen("secv5.in", "rt", stdin);
    freopen("secv5.out", "wt", stdout);

    char buf[32];
    unsigned int tmp;
    int siz;
    scanf("%d%d%d\n", &n, &l, &r);
    for (int i=0; i<n; ++i)
    {
        fgets(buf, 32, stdin);
        tmp = 0; siz = strlen(buf);
        if (buf[siz-1] == '\n') --siz;
        for (int j=0; j<siz; ++j)
            tmp = tmp*10+buf[j]-'0';
        nr[i] = nr2[i] = tmp;
    }

    // normalizare
    sort(nr2, nr2+n);
    int nn = unique(nr2, nr2+n) - nr2;
    for (int i=0; i<n; ++i)
        nr[i] = lower_bound(nr2, nr2+nn, nr[i]) - nr2;

/*
    for (int i=0; i<n; ++i)
        printf("%d ", nr[i]);
    printf("\n");
*/

    // deque
    int ix1=0, ix2=0;
    int dif1=0, dif2=0;
    long long sol = 0;
    for (int i=0; i<n; ++i)
    {
        // min elem
        if (!c1[nr[i]]++) ++dif1;
        while (dif1>=l && ix1<=i)
        {
            if (!--c1[nr[ix1]]) --dif1;
            ++ix1;
        }

        // max elem
        if (!c2[nr[i]]++) ++dif2;
        while (dif2>r && ix2<=i)
        {
            if (!--c2[nr[ix2]]) --dif2;
            ++ix2;
        }

        sol += ix1-ix2;
//        printf("%d %d\n", ix1, ix2);
    }

    printf("%Ld\n", sol);
    return 0;
}