Cod sursa(job #2149803)

Utilizator patcasrarespatcas rares danut patcasrares Data 2 martie 2018 23:39:22
Problema Secventa 5 Scor 90
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.77 kb
#include<fstream>
#include<vector>
#define M 66013
#include<algorithm>
#include<unordered_map>
#include<iostream>
#define pb push_back
#define x first
#define y second
using namespace std;
ifstream fin("secv5.in");
ofstream fout("secv5.out");
unsigned int n,a[(1<<20)+5],p,f,val,poz,st,dr,mij,r1,r2,rez,l,u;
long long ve(int z)
{
    unordered_map<unsigned int,int>mp;
    long long rez=0;
    int j=1;
    for(int i=1;i<=n;i++)
    {
        mp[a[i]]++;
        while(mp.size()>z)
        {
            mp[a[j]]--;
            if(mp[a[j]]==0)
                mp.erase(a[j]);
            j++;
        }
        rez+=(i-j+1);
    }
    return rez;
}
int main()
{
    fin>>n>>l>>u;
    for(int i=1;i<=n;i++)
        fin>>a[i];
    fout<<ve(u)-ve(l-1);
}