Cod sursa(job #2838336)

Utilizator RaresFelixTudose Rares Felix RaresFelix Data 23 ianuarie 2022 13:39:51
Problema Distincte Scor 85
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 3.59 kb
#include <bits/stdc++.h>
#define MN 107171
#pragma GCC optimize("Ofast")
#include <stdio.h>
#include <ctype.h>
//#pragma GCC target("avx2")
using namespace std;
int F[MN], n, k, m, V[MN], R[MN], rez;
tuple<int, int, int> Q[MN];
static inline void add(const int &p) {
    if(!(F[V[p]]++)){
        rez += V[p];
        if(rez > 666012) rez -= 666013;
    }
}
static inline void rem(const int &p) {
    if(!(--F[V[p]])){
         rez -= V[p];
         if(rez < 0) rez += 666013;
    }
}
/** Funcţiile necesare parsării fişierului de intrare **/
FILE *_fin;
int _in_loc; char _in_buff[4096];
void read_init(const char* nume) // Apelaţi această funcţie la începutul funcţiei <main>
{
	_fin = fopen(nume, "r");
	_in_loc = 4095;
}
char read_ch()
{
	++_in_loc;
	if (_in_loc == 4096) {
		_in_loc = 0;
		fread(_in_buff, 1, 4096, _fin);
	}
	return _in_buff[_in_loc];
}
int read_u32() // Apelaţi această funcţie pentru a citi un număr ce se încadrează în categoria <unsigned int>
{
	int u32 = 0; char c;
	while (!isdigit(c = read_ch()) && c != '-');
	int sgn = 1;
	if (c == '-') {
		sgn = -1;
	} else {
		u32 = c - '0';
	}
	while (isdigit(c = read_ch())) {
		u32 = u32 * 10 + c - '0';
	}
	return u32 * sgn;
}
long long read_u64() // Apelaţi această funcţie pentru a citi un număr ce se încadrează în categoria <unsigned long long>
{
	long long u64 = 0; char c;
	while (!isdigit(c = read_ch()) && c != '-');
	long long sgn = 1;
	if (c == '-') {
		sgn = -1;
	} else {
		u64 = c - '0';
	}
	while (isdigit(c = read_ch())) {
		u64 = u64 * 10 + c - '0';
	}
	return u64 * sgn;
}
class OutParser {
private:
    FILE *fout;
    char *buff;
    int sp;
    void write_ch(char ch) {
        if (sp == 50000) {
            fwrite(buff, 1, 50000, fout);
            sp = 0;
            buff[sp++] = ch;
        } else {
            buff[sp++] = ch;
        }
    }
public:
    OutParser(const char* name) {
        fout = fopen(name, "w");
        buff = new char[50000]();
        sp = 0;
    }
    ~OutParser() {
        fwrite(buff, 1, sp, fout);
        fclose(fout);
    }
    OutParser& operator << (int vu32) {
        if (vu32 <= 9) {
            write_ch(vu32 + '0');
        } else {
            (*this) << (vu32 / 10);
            write_ch(vu32 % 10 + '0');
        }
        return *this;
    }
    OutParser& operator << (long long vu64) {
        if (vu64 <= 9) {
            write_ch(vu64 + '0');
        } else {
            (*this) << (vu64 / 10);
            write_ch(vu64 % 10 + '0');
        }
        return *this;
    }
    OutParser& operator << (char ch) {
        write_ch(ch);
        return *this;
    }
    OutParser& operator << (const char *ch) {
        while (*ch) {
            write_ch(*ch);
            ++ch;
        }
        return *this;
    }
} fo("distincte.out");
int main() {
    read_init("distincte.in");
    n = read_u32(), k = read_u32(), m = read_u32();
    for(int i = 1; i <= n; ++i) V[i] = read_u32();
    for(int i = 1; i <= m; ++i) {
        get<0>(Q[i]) = read_u32();
        get<1>(Q[i]) = read_u32();
        get<2>(Q[i]) = i;
    }
    int rad = (int)(.5 + sqrt(double(n)));
    sort(Q+1, Q+m+1, [&](auto a, auto b) {
        if((get<0>(a) / rad) != (get<0>(b) / rad))
            return (get<0>(a) / rad) < (get<0>(b) / rad);
        return bool((get<1>(a) < get<1>(b)) ^ (((get<0>(a) / rad) & 1)));
    });
    int st = 1, dr = 0;
    for(int i = 1; i <= m; ++i) {
        auto [l, r, id] = Q[i];
        while(dr < r) add(++dr);
        while(dr > r) rem(dr--);
        while(st < l) rem(st++);
        while(l < st) add(--st);
        R[id] = rez;
    }
    for(int i = 1; i <= m; ++i) fo << R[i] << "\n";
    return 0;
}