Cod sursa(job #1281267)

Utilizator dragangabrielDragan Andrei Gabriel dragangabriel Data 2 decembrie 2014 23:04:55
Problema Secventa 5 Scor 0
Compilator c Status done
Runda Arhiva de probleme Marime 0.68 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[1100000];

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)
        {
            s[v[j]]--;
            if (!s[v[j]]) s.erase(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;
}