Cod sursa(job #2192388)

Utilizator ScarymovieMocanu Alexandru Scarymovie Data 5 aprilie 2018 20:58:35
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.12 kb
#include <fstream>
#include<bits/stdc++.h>
using namespace std;
const int baza=3;
const int prim=50003;
vector<int> 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%prim].push_back(cod);
    }
    int cod=0,potriviri=0,h=1;
    for(i=1;i<n;++i) h=h*baza;
    for(i=0;i<n;++i)
        cod=cod*baza+text[i]-'a';
    for(;i<l;++i)
    {
        bool ok=0;
        for(unsigned int j=0;j<hashTable[cod%prim].size();++j)
        {
            if(hashTable[cod%prim][j]==cod) ok=1;
            if(ok) {potriviri++;break;}
        }
        cod=cod-(text[i-n]-'a')*h;
        cod=cod*baza+text[i]-'a';
    }
    bool ok=0;
        for(unsigned int j=0;j<hashTable[cod%prim].size();++j)
        {
            if(hashTable[cod%prim][j]==cod) ok=1;
            if(ok) {potriviri++;break;}
        }
    g<<potriviri<<endl;

    return 0;
}