Cod sursa(job #3173227)

Utilizator daria_diaconuDiaconu daria_diaconu Data 22 noiembrie 2023 09:06:18
Problema Secventa 5 Scor 50
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.66 kb
#include <fstream>
#include <unordered_map>
using namespace std;
ifstream cin("secv5.in");
ofstream cout("secv5.out");
int n;
long long v[1050000];
int f(int k)
{
    int ans=0;
    unordered_map<int, int> fr;
    int l=1, cnt=0;
    for(int r=1; r<=n; r++)
    {
	fr[v[r]]++;
        if(fr[v[r]]==1)
            cnt++;
            while(cnt>k)
            {
		fr[v[l]]--;
                if(fr[v[l]]==0)
                    cnt--;
                l++;
            }
        ans+=r-l+1;
    }
    return ans;
}
int main()
{
    int l, u;
    cin>>n>>l>>u;
    for(int i=1; i<=n; i++)
        cin>>v[i];
    cout<<f(u)-f(l-1);
    return 0;
}