Cod sursa(job #917145)

Utilizator razvan9310FMI - Razvan Damachi razvan9310 Data 17 martie 2013 13:15:13
Problema Hashuri Scor 70
Compilator cpp Status done
Runda Arhiva educationala Marime 3.64 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;
        }
		
		const 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);
				int *H2_temp = (int*)malloc(size*sizeof(int));
				memcpy(H2_temp, H2, size);
				
				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);
				free(H1_temp);
				memcpy(H2, H2_temp, size);
				free(H2_temp);
			}
			return *this;
		}
				
		
        const Cuckoo &operator /(int &val)
        {
            printf("%d\n", H1[hashFunction1(val)]==val || H2[hashFunction2(val)]==val);
			return *this;
        }
         
        const Cuckoo &operator -(int &val)
        {
            if (H1[hashFunction1(val)] == val)
                H1[hashFunction1(val)] = 0;
            else 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(1000000);
    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;
        }
    }
    return 0;
}