Cod sursa(job #1573658)

Utilizator codi22FMI Condrea Florin codi22 Data 19 ianuarie 2016 20:40:43
Problema Secventa 5 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 0.77 kb
#include <iostream>
#include <cstdio>
using namespace std;
#include <vector>
#include <unordered_map>
 int V[200000000],n;
 long long int rez(long long int limita)
{
     long long int r=0;
     long long int s,d;
    unordered_map <unsigned int,int> H;
    for (s=1,d=1;d<=n;++d)
    {
        ++H[V[d]];
        while (H.size()>limita)
        {
            //H[V[s]]--;
            if (--H[V[s]]==0) H.erase(V[s]);
            ++s;
        }
        r+=d-s+1;
    }
    return r;
}
int main()
{
   freopen("secv5.in","r",stdin);
   freopen("secv5.out","w",stdout);
   long long int s,d,i;
  // scanf("%lli %lli %lli",&n,&s,&d);
   cin>>n>>s>>d;
   for (i=1;i<=n;i++)
   {
      // scanf("%lli",V[i]);
      cin>>V[i];
   }
   cout<<rez(d)-rez(s-1);
}