Cod sursa(job #1487144)

Utilizator akaprosAna Kapros akapros Data 16 septembrie 2015 10:33:24
Problema Secventa 5 Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.01 kb
#include <cstdio>
#include <cstring>
#include <algorithm>
#include <unordered_map>
#define maxN 1048578
using namespace std;
long long i, j, n, l, u;
long long sol;
unsigned int w[maxN];
unordered_map < unsigned int, unsigned int > H;
void read()
{
    freopen("secv5.in", "r", stdin);
    scanf("%lld %lld %lld", &n, &l, &u);
    for (i = 1; i <= n; ++ i)
        scanf("%ud", &w[i]);
}
long long nrseq(long long u)
{
    long long nr = 1LL;
    long long sol = 0LL;
    H.clear();
    for (i = 1; i <= n; ++ i)
    {
        ++ H[w[i]];
        while (H.size() > u)
        {
            -- H[w[nr]];
            if (!H[w[nr]])
                H.erase(w[nr]);
            ++ nr;
        }
        sol += (i - nr + 1) * 1LL;
    }
    return sol;
}
void solve()
{
    sol = nrseq(u);
    if (l - 1 > 0)
        sol -= nrseq(l - 1);
}
void write()
{
    freopen("secv5.out", "w", stdout);
    printf("%lld", sol);
}
int main()
{
    read();
    solve();
    write();
    return 0;
}