Cod sursa(job #1762717)

Utilizator andreicoman299Coman Andrei andreicoman299 Data 24 septembrie 2016 00:43:26
Problema Secventa 5 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.52 kb
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#define MOD 666013
#define MAXN 1048576

#define BUF_SIZE 16384
char buf[BUF_SIZE];
int pbuf=BUF_SIZE;
FILE*fi,*fo;
inline char nextch(){
    if(pbuf==BUF_SIZE){
        fread(buf, BUF_SIZE, 1, fi);
        pbuf=0;
    }
    return buf[pbuf++];
}
inline int nextnum(){
    int a=0;
    char c=nextch();
    while((c<'0' || c>'9') && c!='-')
        c=nextch();
    int semn=1;
    if(c=='-'){
        semn=-1;
        c=nextch();
    }
    while('0'<=c && c<='9'){
        a=a*10+c-'0';
        c=nextch();
    }
    return a*semn;
}
class Hash{
    public:
    int k=0, next[MAXN+1], *lista;
    int val[MAXN+1];
    int d[MAXN+1];
    Hash() {
        lista = new int[MOD];
        memset(lista, 0x00, MOD*sizeof(int));
    }
    inline void insert(int element, int frecventa){
        k++;
        val[k]=element;
        d[k]=frecventa;
        next[k]=lista[element%MOD];
        lista[element%MOD]=k;
    }
    inline void erase(int element){
        int p=lista[element%MOD];
        if(p==0)
            return ;
        if(val[p]==element){
            d[p]--;
            if(!d[p])
                lista[element%MOD]=next[lista[element%MOD]];
            return ;
        }
        while((next[p]!=0)&&(val[next[p]]!=element))
            p=next[p];
        if(next[p]!=0){
            d[p]--;
            if(!d[p])
                next[p]=next[next[p]];
        }
    }
    inline int find(int element){
        int p=lista[(1LL*element+1LL*MOD*MOD)%MOD];
        while(p!=0 && val[p]!=element)
            p=next[p];
        if(p!=0)
            return p;
        return 0;
    }
} H;

int n;
int v[MAXN+1];
inline long long f(int len){
    int dif=0, first=1;
    long long r=0;
    for(int i=1;i<=n;i++){
        if(!H.find(v[i])){
            H.insert(v[i], 1);
            dif++;
        }
        else
            H.d[H.find(v[i])]++;
        while(dif>len){
            if(H.d[H.find(v[first])]==1)
                dif--;
            H.erase(v[first]);
            first++;
        }
        r+=1LL*i-1LL*first+1LL;
    }
    for(int i=0;i<MOD;i++)
        H.lista[i]=0;
    H.k=0;
    //printf("%d ", r);
    return r;
}

int main(){
    fi=fopen("secv5.in","r");
    fo=fopen("secv5.out","w");
    int l, u;
    n=nextnum(), l=nextnum(), u=nextnum();
    for(int i=1;i<=n;i++)
        v[i]=nextnum();
    fprintf(fo,"%lld", f(u)-f(l-1));
    fclose(fi);
    fclose(fo);
    return 0;
}