Cod sursa(job #2192407)

Utilizator ScarymovieMocanu Alexandru Scarymovie Data 5 aprilie 2018 21:29:31
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.2 kb
#include <fstream>
#include<bits/stdc++.h>
using namespace std;
const long long baza=3;
const long long prim=50003;
vector<long long> hashTable[prim];
string text,cuvant;
long long n,i,j,l,cod,potriviri,idx,p[21];
int main()
{
    ifstream f("abc2.in");
    ofstream g("abc2.out");
    p[0]=1;
    for(i=1;i<=20;++i) p[i]=baza*p[i-1];
    f>>text;
    l=text.length();
    while(f>>cuvant)
    {
        cod=0;
        n=cuvant.length();
        for(i=0; i<n; ++i)
            cod=cod+(cuvant[i]-'a')*p[i];
        hashTable[cod%prim].push_back(cod);
    }
    cod=0,potriviri=0;
    for(i=0; i<n; ++i)
        cod=cod+(text[i]-'a')*p[i];
    for(; i<l; ++i)
    {
        idx=cod%prim;
        for(unsigned int j=0; j<hashTable[idx].size(); ++j)
        {
            if(hashTable[idx][j]==cod)
            {
                potriviri++;
                break;
            }
        }
        cod/=baza;
        cod=cod+(text[i]-'a')*p[n-1];
    }
    idx=cod%prim;
    for(unsigned int j=0; j<hashTable[idx].size(); ++j)
    {
        if(hashTable[idx][j]==cod)
        {
            potriviri++;
            break;
        }
    }
    g<<potriviri<<endl;

    return 0;
}