Cod sursa(job #2192381)

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

    return 0;
}