Cod sursa(job #2871608)

Utilizator Danut200333Dumitru Daniel Danut200333 Data 15 martie 2022 10:23:32
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.02 kb
#include <bits/stdc++.h>

using namespace std;
ifstream fin("abc2.in");
ofstream fout("abc2.out");
#define MOD1 10007
#define hash1 3
string A,B;
vector <int> hashtable[10007];
long long sol,p1,p2,hashB1,hashA1,hashB2,hashA2,n,m;
bool search(int h)
{
    int poz=h%MOD1;
    for(int i=0;i<hashtable[poz].size();i++)if(hashtable[poz][i]==h)return 1;
    return 0;
}
int main()
{
    fin>>A;
    while(fin>>B)
    {
        m=B.size();
        hashB1=0;
        for(int i=0; i<m; i++)
        {
            hashB1=((hashB1*hash1)+(B[i]-'a'));
        }
        if(!search(hashB1))hashtable[hashB1%MOD1].push_back(hashB1);
    }
    p1=p2=1;
    n=A.size();
    for(int i=0; i<m; i++)
    {
        hashA1=((hashA1*hash1)+(A[i]-'a'));
        if(i>0)
        {
            p1=(p1*hash1);
        }
    }
    if(search(hashA1))sol++;
    for(int i=m; i<n; i++)
    {
        hashA1=((hash1*(hashA1-((A[i-m]-'a')*p1))))+(A[i]-'a');
        if(search(hashA1))sol++;
    }
    fout<<sol;
    return 0;
}