Cod sursa(job #1036290)

Utilizator cat_red20Vasile Ioana cat_red20 Data 19 noiembrie 2013 10:02:25
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.15 kb
#include<fstream>
#include<algorithm>
#include<cstring>
#include<vector>
#define MOD 666013

using namespace std;
int l,lung,k,u,poz,sol;
unsigned int p3,val;
char s[10000001],sir[21];
vector<unsigned int> v[MOD+1];
ifstream fin("abc2.in");
ofstream fout("abc2.out");

unsigned int genereazaHash(char s[],int l)
{
	unsigned int h=0;
	for(int i=0;i<l;i++)
	{
		h=h*3+s[i]-'a';
	}
    return h;
}

void prelucareDictionar()
{
    unsigned int h;
	fin.getline(sir,21);
	lung=strlen(sir);
	do
	{
		h=genereazaHash(sir,lung);
		v[h%MOD].push_back(h);
	}while(fin.getline(sir,21));
}

void putere3()
{
	p3=1;
	for(int i=1;i<lung;i++)
	{
		p3*=3;
	}
}

int cauta(unsigned int val)
{
    unsigned int k=val%MOD;
    for(int i=0;i<v[k].size();i++)
    {
        if(v[k][i]==val)
        return 1;
    }
    return 0;
}

int main()
{
	fin.getline(s,10000001);
	l=strlen(s);
	prelucareDictionar();
	if(l<lung)
	{
		fout<<0;
		return 0;
	}
	putere3();
	val=genereazaHash(s,lung);
	sol+=cauta(val);
	for(int i=lung;i<l;i++)
	{
        val=(val%p3)*3+s[i]-'a';
        sol+=cauta(val);
	}
	fout<<sol;
	return 0;
}