Cod sursa(job #3171073)

Utilizator stefanvilcescuStefan Vilcescu stefanvilcescu Data 18 noiembrie 2023 12:41:00
Problema Secventa 5 Scor 0
Compilator cpp-64 Status done
Runda Teme Pregatire ACM Unibuc 2014, Anul I Marime 0.84 kb
#include <bits/stdc++.h>
#include <unordered_map>

using namespace std;
const int MAXN=1e7;

int sir[MAXN];
int n;

long long cnt(int x){
    int i, j, nr;
    unordered_map<int, int> frecv;
    long long ans;
    i=j=nr=ans=0;
    while(j<n){
        if(j<n && nr<=x){
            frecv[sir[j]]++;
            if(frecv[sir[j]]==1)
                nr++;
            j++;
        }else{
            while(nr==x+1){
                frecv[sir[i]]--;
                if(frecv[sir[i]]==0)
                    nr--;
                i++;
            }
        }
        if(nr<=x)
            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;
}