Cod sursa(job #1280405)

Utilizator dragangabrielDragan Andrei Gabriel dragangabriel Data 1 decembrie 2014 22:04:35
Problema Secventa 5 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.69 kb
#include<cstdio>
#include<algorithm>
#include<unordered_map>
using namespace std;

unordered_map<int,int>s;
int n,i,j,k,lg,x,y,v[110000];

long long solve(int x)
{
    s.clear();
    long long rez=0;
    int j=1;
    for(int i=1;i<=n;i++)
    {
        s[v[i]]++;
        while(s.size()>x)
        {
            if (s[v[j]]==1) s.erase(v[j]);else
                s[v[j]]--;
            j++;
        }
        rez=rez+i-j+1;
    }
    return rez;
}

int main()
{
    freopen("secv5.in","r",stdin);
    freopen("secv5.out","w",stdout);
    scanf("%d %d %d",&n,&x,&y);
    for(i=1;i<=n;i++)
        scanf("%d",&v[i]);
    printf("%d\n",solve(y)-solve(x-1));
    return 0;
}