Cod sursa(job #1478489)

Utilizator retrogradLucian Bicsi retrograd Data 28 august 2015 19:23:52
Problema Hashuri Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.97 kb
#include <cstdio>
#include <cstring>
#include <algorithm>

using namespace std;
typedef int var;

struct HMap {
    int P;

    #define val first
    #define nxt second

    pair<int, int> T[1000005];
    int *B, *Size, nodes = 0;

    HMap(int P) {
        this->P = P;

        B = new int[P];
        Size = new int[P];

        memset(B, 0, sizeof(B));
        memset(Size, 0, sizeof(Size));
    }

    int find(int x) {
        for(int y=B[x % P]; y; y = T[y].nxt)
            if(T[y].val == x)
                return y;
        return 0;
    }

    void erase(int x) {
        int y = find(x);
        if(y) T[y].val = 0;
    }

    void insert(int x) {
        int y = find(x);
        if(!y) {
            T[++nodes] = {x, B[x % P]};
            B[x % P] = nodes;
            Size[x % P]++;
        }
    }
};

#define P1 65536
#define P2 42113
HMap H1(P1), H2(P2);

void ins(int x) {
    if(H1.Size[x % P1] < H2.Size[x % P2])
        H1.insert(x);
    else
        H2.insert(x);
}

void ers(int x) {
    H1.erase(x);
    H2.erase(x);
}

bool fnd(int x) {
    if(H1.find(x) || H2.find(x)) return 1;
    return 0;
}

#define isdigit(c) (c>='0'&&c<='9')
struct Scanner {
    #define DIM 100000
    char buff[DIM];
    int poz = DIM - 1;
    bool sign;

    Scanner& operator >> (var &a) {
        #define adv() (++poz == DIM) ? fread(buff, 1, DIM, stdin), poz=0 : 0
        for(sign = 0; !isdigit(buff[poz]) && buff[poz] != '-'; adv());
        if(buff[poz] == '-') sign = 1, adv();
        for(a = 0; isdigit(buff[poz]); a = a * 10 + buff[poz] - '0', adv());
        if(sign) a = -a;

        return *this;
    }
};
Scanner in;

int main() {
    freopen("hashuri.in", "r", stdin);
    freopen("hashuri.out", "w", stdout);

    int m, t, a;
    in >> m;
    while(m--) {
        in >> t >> a;
        if(t == 1) ins(a);
        if(t == 2) ers(a);
        if(t == 3) printf("%d\n", fnd(a));
    }

    return 0;
}