Cod sursa(job #988731)

Utilizator maritimCristian Lambru maritim Data 23 august 2013 19:00:51
Problema Secventa 5 Scor 80
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.87 kb
#include<stdio.h>
#include<algorithm>
#include<vector>
#include<fstream>
#include<ctype.h>
using namespace std;

ifstream f("secv5.in");
ofstream g("secv5.out");

typedef vector<pair<unsigned int,int> >::iterator it;

#define MaxN ((1<<20)+100)
#define M 2013
#define ll long long
#define ui unsigned int

int N,L,U,l,u,poz_l=1,poz_u = 1;
ll Sol;
ui A[MaxN],B[MaxN];
int BestL[MaxN],BestU[MaxN];
vector<pair<ui,int> > H[M];
char S[10*MaxN];

void citire(void)
{
    int a = 0;

    f >> N >> U >> L;
    f.getline(S,10*MaxN,EOF);
    for(int i=0;S[i];i++)
        if(isdigit(S[i]))
            for(A[++a] = 0;isdigit(S[i]);A[a] = A[a]*10+S[i++]-'0');
    -- U;
}

inline void AddHash(ui a,int poz_norm)
{
    int x = a%M;

    for(it i=H[x].begin();i<H[x].end();i++)
        if(i->first == a)
            return ;

    H[x].push_back(make_pair(a,poz_norm));
}

inline int ValHash(ui a)
{
    int x = a%M;

    for(it i=H[x].begin();i<H[x].end();i++)
        if(i->first == a)
            return i->second;

    return -1;
}

void Normalizare(void)
{
    int poz = 0;
    
    for(int i=1;i<=N;i++)
        B[i] = A[i];

    sort(B+1,B+N+1);

    for(int i=1;i<=N;i++)
        if(B[i] != B[i-1])
            AddHash(B[i],++poz);

    for(int i=1;i<=N;i++)
        A[i] = ValHash(A[i]);
}

inline void AdaugareL(int a)
{
    l += (!BestL[a]);

    BestL[a]++;

    while(l>L)
    { 
        a = A[poz_l];

        BestL[a]--; poz_l ++;

        if(!BestL[a])
            -- l;
    }
}

inline void AdaugareU(int a)
{
    u += (!BestU[a]);

    BestU[a] ++;

    while(u>U)
    {
        a = A[poz_u];

        BestU[a] --; poz_u ++;

        if(!BestU[a])
            --u;
    }
}

void Rezolvare(void)
{
    Normalizare();

    for(int i=1;i<=N;i++)
    {
        AdaugareL(A[i]);
        AdaugareU(A[i]);
        Sol += 1LL*poz_u-poz_l;
    }
}

int main()
{
    citire();
    Rezolvare();

    g << Sol << "\n";
}