Cod sursa(job #2265673)

Utilizator Moise_AndreiMoise Andrei Moise_Andrei Data 21 octombrie 2018 15:45:33
Problema Secventa 5 Scor 80
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.71 kb
#include <bits/stdc++.h>
#define int long long
using namespace std;
ifstream in("secv5.in");
ofstream out("secv5.out");
int v[1000005];
int f(int x, int n)
{
    unordered_map <int, int> mp;
    int nr = 0, cnt = 1;
    for(int i = 1; i <= n; i++)
    {
        mp[v[i]]++;
        while(mp.size() > x)
        {
            mp[v[cnt]]--;
            if(mp[v[cnt]] == 0)
                mp.erase(v[cnt]);
            cnt++;
        }
        nr += (i - cnt + 1);
       // cout << nr << " ";
    }
   // cout << '\n';
    return nr;
}
int32_t main()
{
    int n, a, b;
    in >> n >> a >> b;
    for(int i = 1; i <= n; i++)
        in >> v[i];
    out << f(b, n) - f(a - 1, n);
    return 0;
}