Cod sursa(job #837520)

Utilizator szabibibiOrban Szabolcs szabibibi Data 18 decembrie 2012 02:14:53
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.75 kb
#include <stdio.h>
#include <vector>
#include <string.h>

#define CMAX 10000002
#define NMAX 50000
#define LMAX 22
#define P 666013

using namespace std;

char cuv[CMAX];
vector <unsigned long> hash[P];
unsigned long harr;
long l, c;
long count;

unsigned long val(char *s)
{
    long i;
    unsigned long sol = 0;   
    for (i = l - 1; i >= 0; i--)
        sol = (sol * 3) + (s[i] - 'a');
    return sol;
}

void read()
{
     freopen("abc2.in", "r", stdin);
     char s[LMAX];
     unsigned long vall;
     
     gets(cuv);
     c = strlen(cuv);

     gets(s);
     l = strlen(s);

     vall = val(s);
     hash[vall % P].push_back(vall);

     while (gets(s))
     {
           vall = val(s);
           hash[vall % P].push_back(vall);
     }
}

unsigned long calc_put()
{
     int i;
     unsigned long p = 1;
     for (i = 1; i < l; i++)
         p = p * 3;
     return p;
}

int search_val(unsigned long vall)
{
    unsigned long poz = vall % P;
    vector <unsigned long> :: iterator ittt;

    for (ittt = hash[poz].begin(); ittt != hash[poz].end(); ++ittt)
        if (*ittt == vall)
           return 1;
    return 0;
}

void dostuff()
{
     long i;
     unsigned long vall = val(cuv), sval, eval;

     if (search_val(vall))
        count++;
        
     harr = calc_put();
     
     for (i = 1; i <= c - l; i++)
     {
         sval = (cuv[i - 1] - 'a') * harr;
         eval = cuv[i + l - 1] - 'a';
         vall = (vall - sval) * 3 + eval;
         if (search_val(vall))
            count++;
     }
}

void print()
{
     FILE *f = fopen("abc2.out", "w");
     fprintf(f, "%ld", count);
     fclose(f);
}

int main()
{
    read();
    dostuff();
    print();
    return 0;
}