Cod sursa(job #2715582)

Utilizator beingsebiPopa Sebastian beingsebi Data 3 martie 2021 21:20:01
Problema Distincte Scor 75
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.82 kb
#include <bits/stdc++.h>
using namespace std;
ifstream f("distincte.in");
ofstream g("distincte.out");
const int mod = 666013;
int n, m, vmax, v[100001], cnt[100001];
struct query
{
    int st, dr, p;
} q[100001];
void add(int);
void remove(int);
int rez, af[100001];
void change(int, int, int, int);
int main()
{
    f >> n >> vmax >> m;
    int dim = sqrt(n);
    for (int i = 1; i <= n; i++)
        f >> v[i];
    for (int i = 1; i <= m; i++)
        f >> q[i].st >> q[i].dr, q[i].p = i;
    sort(q + 1, q + m + 1, [dim](query a, query b) {
        int ax = a.st / dim, ay = b.st / dim;
        if (ax != ay)
            return ax < ay;

        return a.dr > b.dr;
    });
    int st = 1, dr = 1;
    rez = v[1];
    cnt[v[1]] = 1;
    for (int i = 1; i <= m; i++)
    {
        int nst = q[i].st, ndr = q[i].dr;
        change(st, dr, nst, ndr);
        st = nst;
        dr = ndr;
        af[q[i].p] = rez;
    }
    for (int i = 1; i <= m; i++)
        g << af[i] << '\n';
    return 0;
}
void add(int x)
{
    cnt[x]++;
    if (cnt[x] == 1)
        rez += x;
    if (rez >= mod)
        rez -= mod;
}
void remove(int x)
{
    cnt[x]--;
    if (cnt[x] == 0)
        rez -= x;
    if (rez < 0)
        rez += mod;
}
void change(int st, int dr, int nst, int ndr)
{
    if (dr <= nst || st >= ndr)
    {
        for (int i = st; i <= dr; i++)
            remove(v[i]);
        for (int i = nst; i <= ndr; i++)
            add(v[i]);
        return;
    }
    if (st < nst)
        for (int i = st; i < nst; i++)
            remove(v[i]);
    else
        for (int i = st - 1; i >= nst; i--)
            add(v[i]);
    if (dr < ndr)
        for (int i = dr + 1; i <= ndr; i++)
            add(v[i]);
    else
        for (int i = dr; i > ndr; i--)
            remove(v[i]);
}