Cod sursa(job #2260541)

Utilizator tanasaradutanasaradu tanasaradu Data 15 octombrie 2018 09:30:10
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.52 kb
#include <bits/stdc++.h>

using namespace std;


ifstream fin ("abc2.in");
ofstream fout ("abc2.out");

const int Nmax =  10000005;
const int P = 50003;
const int P1 = 12347;

char sir[Nmax + 5];

char a[55];

int La, Lsir;


vector < unsigned int > L[2 * P];


inline bool Ok(unsigned int H, unsigned int H1)
{
    int r;
    r = H % P;
    for(auto it : L[r])
        if(it == H1)
            return true;
    return false;
}

int main()
{
    int sol = 0;
    unsigned int p, H, H1, p1;
    fin >> (sir + 1);
    Lsir = strlen(sir + 1);
    while(fin >> (a + 1))
    {
        La = strlen(a + 1);
        H = H1 = 0;
        for(int j = 1 ; j <= La ; j++)
        {
            H = (H * 3 + (a[j] - 'a'));
            H1 = (H1 * 3 + (a[j] - 'a'));
        }
        if(!Ok(H, H1))
            L[H % P] . push_back(H1 % P1);
    }
    if(La > Lsir)
    {
        fout << "0\n";
        return 0;
    }

    H = H1 = 0;
    p = p1 = 1;
    for(int i = 1 ; i <= La ; i++)
    {
        H = (H * 3 + (sir[i] - 'a'));
        H1 = (H1 * 3 + (sir[i] - 'a'));
        if(i > 1)
        {
            p = (p * 3);
            p1 = (p1 * 3);
        }
    }
    sol += Ok(H, H1 % P1);
    for(int i = La + 1; i <= Lsir ; i++)
    {
        H = ((H - (sir[i - La] - 'a') * p) * 3 + (sir[i] - 'a'));
        H1 = ((H1 - (sir[i - La] - 'a') * p1) * 3 + (sir[i] - 'a'));
        sol += Ok(H, H1 % P1);
    }
    fout << sol << "\n";
    fin.close();
    fout.close();
    return 0;
}