Cod sursa(job #917283)

Utilizator razvan9310FMI - Razvan Damachi razvan9310 Data 17 martie 2013 16:06:18
Problema Hashuri Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 4.19 kb
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <ctime>
#define prim 1000000009
 
class Cuckoo
{
    private:
        int a1, a2, b1, b2;
        int size, *H1, *H2;
          
    public:
        Cuckoo(int SIZE)
        {
            srand(time(NULL));
            size = SIZE;
            H1 = (int*)calloc(size, sizeof(int));
            H2 = (int*)calloc(size, sizeof(int));
            a1=rand()%size; a2=rand()%size;
            b1=rand()%size; b2=rand()%size;
        }           
          
        int hashFunction1(int &x)
        {
            return (((long long)a1*x + b1)%prim)%size;
        }
          
        int hashFunction2(int &x)
        {
            return (((long long)a2*x + b2)%prim)%size;
        }
         
        int hash(int x)
        {
            if (H1[hashFunction1(x)]==x || H2[hashFunction2(x)]==x)
                return -1;
            int aux, val1, val2, i, old=x;
            bool inPlace = 0;
            for (i=0; !inPlace; ++i)
            {
                if (i && x==old)
                    return 0;
                val1=hashFunction1(x); val2=hashFunction2(x);
                if (H1[val1] == 0)
                {
                    H1[val1] = x;
                    inPlace = 1;
                }
                else if (H2[val2] == 0)
                {
                    H2[val2] = x;
                    inPlace = 1;
                }
                else
                {
                    aux = H1[val1];
                    H1[val1]=x; x=aux;
                    val2 = hashFunction2(x);
                    if (H2[val2] == 0)
                    {
                        H2[val2] = x;
                        inPlace = 1;
                    }
                    else
                    {
                        aux = H2[val2];
                        H2[val2]=x; x=aux;
                    }
                }
            }
            return 1;
        }
         
        Cuckoo &operator +(int val)
        {
            int canInsert = hash(val);
            if (!canInsert) //ciclu => rehash
            {
                int *H1_temp = (int*)malloc(size*sizeof(int));
                memcpy(H1_temp, H1, size*sizeof(int));
                int *H2_temp = (int*)malloc(size*sizeof(int));
                memcpy(H2_temp, H2, size*sizeof(int));
                 
                while (!canInsert)
                {
                    a1=rand()%size; a2=rand()%size;
                    b1=rand()%size; b2=rand()%size;
                    free(H1); free(H2);
                    H1 = (int*)calloc(size, sizeof(int));
                    H2 = (int*)calloc(size, sizeof(int));
                     
                    canInsert = 1;
                    for (int i=0; i<size && canInsert==1; ++i)
                    {
                        if (H1_temp[i])
                            canInsert = hash(H1_temp[i]);
                        if (canInsert==1 && H2_temp[i])
                            canInsert = hash(H2_temp[i]);
                    }
                }
                 
                //memcpy(H1, H1_temp, size*sizeof(int));
                free(H1_temp);
                //memcpy(H2, H2_temp, size*sizeof(int));
                free(H2_temp);
            }
            return *this;
        }
         
        Cuckoo &operator ^(int val)
        {
            printf("%d\n", H1[hashFunction1(val)]==val || H2[hashFunction2(val)]==val);
            return *this;
        }
          
        Cuckoo &operator -(int val)
        {
            if (H1[hashFunction1(val)] == val)
                H1[hashFunction1(val)] = 0;
            else if (H2[hashFunction2(val)] == 0)
				H2[hashFunction2(val)] = 0;
            return *this;
        }
};
 
int main()
{
    freopen("hashuri.in", "r", stdin);
    freopen("hashuri.out", "w", stdout);
    int n, c, x, i;
    Cuckoo H(750000);
    scanf("%d", &n);
    for (i=0; i<n; ++i)
    {
        scanf("%d%d", &c, &x);
        switch (c)
        {
            case 1: H+x; break;
            case 2: H-x; break;
            case 3: H^x; break;
        }
    }
	//H-1^1+1^1;
    return 0;
}