Cod sursa(job #2296808)

Utilizator pasoi_stefanPasoi Stefan pasoi_stefan Data 5 decembrie 2018 00:50:14
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.74 kb
#include<fstream>
#include<string>
#include<vector>
#include<set>
#define MOD 10041
using namespace std;
ifstream cin("abc2.in");
ofstream cout("abc2.out");
string a,b;
set<unsigned> Hash[MOD+1];
int n,m,k;
unsigned ans,code,p;
int main(){
    cin>>a;
    while(cin>>b){
        code=0;
        for(int i=0;i<b.size();i++)
            code=code*3+(b[i]-'a');
        Hash[code%MOD].insert(code);
    }
    k=b.size(); p=1;
    for(int i=1;i<k;i++)
        p*=3;
    code=0;
    for(int i=0;i<k;i++)
        code=code*3+(a[i]-'a');
    ans+=Hash[code%MOD].count(code);

    for(int i=k;i<a.size();i++){
        code=(code-(a[i-k]-'a')*p)*3+(a[i]-'a');
        ans+=Hash[code%MOD].count(code);
    }

    cout<<ans;
}