Cod sursa(job #2647306)

Utilizator CONFUSION@@@@@ @@@ CONFUSION Data 3 septembrie 2020 21:08:18
Problema Secventa 5 Scor 80
Compilator cpp-64 Status done
Runda Teme Pregatire ACM Unibuc 2014, Anul I Marime 1.26 kb
#include <bits/stdc++.h>
#define int unsigned long long // imi bag pula
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);
    }
};

vector <int> v;
unordered_map <int, int, custom_hash> Hash;

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

    auto poz = v.begin();
    for (auto it = v.begin(); it != v.end(); ++it) {
        ++Hash[*it];
        while (Hash.size() > x) {
            if (--Hash[*poz] == 0)
                Hash.erase(*poz);
            poz++;
        }
        
        ans += it - poz + 1;
    }

    return ans;
}

signed main () {
    freopen("secv5.in", "r", stdin);
    freopen("secv5.out", "w", stdout);

    ios::sync_with_stdio(false);
    cin.tie(nullptr);
    cout.tie(nullptr);

    cin >> n >> l >> u;
    v.resize(n);
    for (auto &it: v)
        cin >> it;

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