Cod sursa(job #907441)

Utilizator mariacMaria Constantin mariac Data 7 martie 2013 22:37:36
Problema Hashuri Scor 30
Compilator cpp Status done
Runda Arhiva educationala Marime 3.2 kb
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <time.h>
using namespace std;

unsigned int* aloca(unsigned int *h, int MOD)
{
    int i;
    free(h);
    h = (unsigned int *) malloc (MOD * sizeof(long long));
    for (i = 0; i < MOD; i++)
        h[i] = 0;
    return h;
}

void SetHash (int &a, int &b, int &c, int &MOD)
{
    int mo[]={999983,999979,899981,900007,800011};
    a = rand () % 100 + 1;
    b = rand () % 100 + 1;
    c = rand () % 100 + 1;
    MOD = mo[rand () % 5];
}

int insereaza (unsigned int x, int a, int b, int c, unsigned int *h1, unsigned int *h2, int MOD)
{
    int ok = 0, st = 1, v1, v2;
    unsigned int aux;

    v1 = (a * (x %MOD)) % MOD;
    v2 = ((b * (x %MOD)) % MOD + c * c)%MOD;

    while( ok > -50 )
    {
        if( st == 1 )
        {
            if(!h1 [v1] || h1 [v1] == x)
            {
                h1 [v1] = x;
                return 1;
            }
            else
            {
                st = -1;
                if( ok < 0)
                {
                    aux = x;
                    x =  h1 [v1];
                    h1 [v1] = aux;
                }
            }
        }
        if( st == -1)
        {
            if(!h2 [v2] || h2 [v2] == x)
            {

                h2 [v2] = x;

                return 1;
            }

            else
            {
                st = 1;
                if( ok < 0)
                {
                    aux = h2 [v2];
                    h2 [v2] = x;
                    x = aux;
                }
            }
        }
        ok--;
    }
    return 0;
}
void del(unsigned int x, int a, int b, int c, unsigned int *h1, unsigned int *h2, int MOD)
{
    int v1,v2;
    v1 = (a * (x %MOD)) % MOD;
    v2 = ((b * (x %MOD)) % MOD + c * c)%MOD;
    if(h1[v1] == x) h1[v1] = 0;

        else if(h2[v2] == x ) h2[v2] = 0;
}

int cauta(unsigned int x, int a, int b, int c, unsigned int *h1, unsigned int *h2, int MOD)
{
     if(h1 [(a * (x % MOD)) % MOD] == x) return 1;

     if(h2 [((b * (x % MOD)) % MOD + c * c) % MOD] == x ) return 1;

    return 0;
}

void citeste (int a, int b, int c, unsigned int *h1, unsigned int *h2, int MOD)
{
    FILE *f = fopen ("hashuri.in", "r");
    FILE *fi = fopen ( "hashuri.out", "w");
    int n, caz, i;
    unsigned int x;

    fscanf (f,"%d", &n);

    for (i = 1; i <= n; i++)
    {
        fscanf (f, "%d %d", &caz, &x);

        switch (caz)
        {
            case 1: while(!insereaza( x, a, b, c, h1, h2, MOD))
                    {
                        SetHash (a, b, c, MOD);
                        aloca(h1, MOD);
                        aloca(h2, MOD);
                        citeste (a, b, c, h1, h2, MOD);
                    }
                    break;
           case 2: del(x, a, b, c, h1, h2, MOD); break;
           case 3: fprintf( fi,"%d\n", cauta(x, a, b, c, h1, h2, MOD)); break;

        }
    }
}




int main()
{
    int a, b, c, MOD;
    unsigned int *h1, *h2;

    srand (time(NULL));

    SetHash (a, b, c, MOD);
    h1 = aloca(h1, MOD);
    h2 = aloca(h2, MOD);
    citeste (a, b, c, h1, h2, MOD);

    return 0;

}