Cod sursa(job #2647309)

Utilizator CONFUSION@@@@@ @@@ CONFUSION Data 3 septembrie 2020 21:13:11
Problema Secventa 5 Scor 80
Compilator cpp-64 Status done
Runda Teme Pregatire ACM Unibuc 2014, Anul I Marime 1.13 kb
#include <bits/stdc++.h>
#define int unsigned long long
using namespace std;

struct custom_hash {
    static uint64_t splitmix64 (uint64_t x) {
        x += 0x9e3779b97f4a7c15;
        x = (x ^ (x >> 30)) * 0xbf58476d1ce4e5b9;
        x = (x ^ (x >> 27)) * 0x94d049bb133111eb;
        return x ^ (x >> 31);
    }

    size_t operator () (uint64_t x) const {
        static const uint64_t FIXED_RANDOM = chrono::steady_clock::now().time_since_epoch().count();
        return splitmix64(x + FIXED_RANDOM);
    }
};

unordered_map <int, int, custom_hash> Hash;

int *v, n, l, u;
int go (int x) {
    Hash.clear();
    int ans = 0;

    int poz = 1, it;
    for (it = 1; it <= n; it++) {
        ++Hash[v[it]];
        while (Hash.size() > x) {
            if (--Hash[v[poz]] == 0)
                Hash.erase(v[poz]);
            poz++;
        }
        
        ans += it - poz + 1;
    }

    return ans;
}

signed main () {
    ifstream fin ("secv5.in");
    ofstream fout ("secv5.out");

    fin >> n >> l >> u;
    v = new (nothrow) int[n + 1];
    
    int i;
    for (i = 1; i <= n; i++)
        fin >> v[i];

    fout << go(u) - go(l - 1);
    return 0;
}