Cod sursa(job #2953358)

Utilizator AndreibatmanAndrei Croitoriu Andreibatman Data 11 decembrie 2022 01:28:24
Problema Secventa 5 Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.67 kb
#include <bits/stdc++.h>
#define int long long
using namespace std;
ifstream fin("secv5.in");
ofstream fout("secv5.out");
unordered_map<int,int>fr;
int ans,st,dr,nr,v[1000010],i,l,r,n;
int solve(int x)
{
    int st=1,ans=0,nr=0;
    for(int i=1;i<=n;i++)
    {
        if(fr[v[i]]==0)
            nr++;
        fr[v[i]]++;
        while(st<=i && nr>x)
        {
            fr[v[st]]--;
            if(fr[v[st]]==0)
                nr--;
            st++;
        }
        ans=ans+i-st+1;
    }
    fr.clear();
    return ans;
}
signed main()
{
    fin>>n>>l>>r;
    for(i=1; i<=n; i++)
        fin>>v[i];
    fout<<solve(r)-solve(l-1);
    return 0;
}