Cod sursa(job #1786953)

Utilizator GeanaVladGeana Vlad GeanaVlad Data 23 octombrie 2016 21:16:33
Problema Secventa 5 Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.27 kb
#include <iostream>
#include<vector>
#include<fstream>
#define MOD 666013
using namespace std;
ifstream f("secv5.in");
ofstream g("secv5.out");
long long n,a,b,i,x,v[1200010],nr,vf[1200000],L[1200010];
vector<pair<long long,long long> >h[MOD];
long long exista(long long x)
{
    long long m=x%MOD;
    int sz=h[m].size();
    for(long long i=0;i<sz;i++)
        if(h[m][i].first==x)
            return h[m][i].second;
    return 0;
}
long long normalizare(long long x)
{
    long long p=exista(x);
    if(p)
        return p;
    else
    {
        nr++;
        h[x%MOD].push_back(make_pair(x,nr));
        return nr;
    }
}
long long procedura(long long a)
{
    long long i;
    for(i=1;i<=n;i++)
        vf[v[i]]=0;
    long long p=1,nr=0,sum=0;
    for(i=1;i<=n;i++)
    {
        vf[v[i]]++;
        if(vf[v[i]]==1)
        {
            nr++;
            while(nr>a)
            {
                vf[v[p]]--;
                if(vf[v[p]]==0)
                    nr--;
                p++;
            }
        }
        L[i]=p;
        sum=sum+(i-L[i]+1);
    }
    return sum;
}
int main()
{
    f>>n>>a>>b;
    for(i=1;i<=n;i++)
    {
        f>>x;
        v[i]=normalizare(x);
    }
    g<<procedura(b)-procedura(a-1);
}