Cod sursa(job #2432771)

Utilizator IulianOleniucIulian Oleniuc IulianOleniuc Data 24 iunie 2019 23:46:59
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.93 kb
#include <bits/stdc++.h>
using namespace std;

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

const int BASE = 3;
unsigned int getHashCode(string str) {
    unsigned int h = 0;
    for (int i = 0; i < (int) str.size(); i++)
        h = h * BASE + (str[i] - 'a');
    return h;
}

int main() {
    string str; fin >> str;
    unordered_set<unsigned int> patterns; int len = 0;

    string pattern;
    while (fin >> pattern) {
        len = pattern.size();
        patterns.insert(getHashCode(pattern));
    }

    unsigned int h = getHashCode(str.substr(0, len));
    int sol = (patterns.find(h) != patterns.end());

    unsigned int pwr = 1;
    for (int i = 1; i < len; i++)
        pwr *= BASE;
    
    for (int i = len; i < (int) str.size(); i++) {
        h = (h - (str[i - len] - 'a') * pwr) * BASE + (str[i] - 'a');
        sol += (patterns.find(h) != patterns.end());
    }
    fout << sol << '\n';

    fout.close();
    return 0;
}