Cod sursa(job #1768005)

Utilizator oldatlantianSerban Cercelescu oldatlantian Data 29 septembrie 2016 23:46:03
Problema Distincte Scor 100
Compilator cpp Status done
Runda Arhiva de probleme Marime 2.67 kb
#include <bits/stdc++.h>
using namespace std;
typedef long long i64;

const int NMAX = 100005,
           MOD = 666013;

class InParser {
private:
    FILE *fin;
    char *buff;
    int sp;

    char read_ch() {
        ++sp;
        if (sp == 4096) {
            sp = 0;
            fread(buff, 1, 4096, fin);
        }
        return buff[sp];
    }

public:
    InParser(const char* nume) {
        fin = fopen(nume, "r");
        buff = new char[4096]();
        sp = 4095;
    }

    InParser& operator >> (int &n) {
        char c;
        while (!isdigit(c = read_ch()) && c != '-');
        int sgn = 1;
        if (c == '-') {
            n = 0;
            sgn = -1;
        } else {
            n = c - '0';
        }
        while (isdigit(c = read_ch())) {
            n = 10 * n + c - '0';
        }
        n *= sgn;
        return *this;
    }

    InParser& operator >> (long long &n) {
        char c;
        n = 0;
        while (!isdigit(c = read_ch()) && c != '-');
        long long sgn = 1;
        if (c == '-') {
            n = 0;
            sgn = -1;
        } else {
            n = c - '0';
        }
        while (isdigit(c = read_ch())) {
            n = 10 * n + c - '0';
        }
        n *= sgn;
        return *this;
    }
};

struct QUERY {
    int st, dr, index;
    i64 ant;
};

int n, k, m;

int        nxt[NMAX],
           lst[NMAX];
i64        aib[NMAX],
             v[NMAX];
QUERY        q[NMAX];

bool cmp_a(const QUERY &a, const QUERY &b) {
    return a.st < b.st;
}

bool cmp_b(const QUERY &a, const QUERY &b) {
    return a.index < b.index;
}

inline int lsb(int arg) {
    return arg&-arg;
}

inline i64 aib_query(int pos) {
    i64 ant = 0;
    while(pos > 0) {
        ant = ant + aib[pos];
        pos-= lsb(pos);
    }
    return ant;
}

inline void aib_update(int pos, i64 val) {
    while(pos < NMAX) {
        aib[pos]+= val;
        pos+= lsb(pos);
    }
}

int main(void) {
    InParser fi("distincte.in");
    ofstream fo("distincte.out");
    int st;

    fi >> n >> k >> m;
    for(int i=1; i<=n; ++i)
        fi >> v[i];

    for(int i=1; i<=m; ++i) {
        fi >> q[i].st >> q[i].dr; //csf...
        q[i].index = i;
        q[i].ant = 0;
    }

    for(int i=1; i<=n; ++i) {
        nxt[lst[v[i]]] = i;
        if(!lst[v[i]])
            aib_update(i, v[i]);
        lst[v[i]] = i;
    }

    sort(q+1, q+1+m, cmp_a);

    st = 1;
    for(int i=1; i<=m; ++i) {
        for(; st<q[i].st; ++st) {
            aib_update(st, -v[st]);
            if(nxt[st])
                aib_update(nxt[st], v[st]);
        }
        q[i].ant = aib_query(q[i].dr);
    }

    sort(q+1, q+1+m, cmp_b);

    for(int i=1; i<=m; ++i)
        fo << q[i].ant % MOD << '\n';

    return 0;
}