Cod sursa(job #3185775)

Utilizator cosminteoaTeodorescu Cosmin cosminteoa Data 20 decembrie 2023 13:25:11
Problema Secventa 5 Scor 100
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.73 kb
#include <bits/stdc++.h>
#include <unordered_map>

using namespace std;
const int MAXN=1e7;

long long sir[MAXN];
int n;

long long cnt(int x){
    int i, j, nr;
    unordered_map<long long, int> frecv;
    long long ans;
    i=j=nr=ans=0;
    while(j<n){
        frecv[sir[j]]++;
        if(frecv[sir[j]]==1)
            nr++;
        j++;
        while(nr>x){
            frecv[sir[i]]--;
            if(frecv[sir[i]]==0)
                nr--;
            i++;
        }
        ans+=j-i;
    }
    return ans;
}

int main(){
    ifstream fin("secv5.in");
    ofstream fout("secv5.out");
    int i, l, u;
    fin>>n>>l>>u;
    for(i=0; i<n; i++)
        fin>>sir[i];
    fout<<cnt(u)-cnt(l-1);
    return 0;
}