Cod sursa(job #3266650)

Utilizator Barbu_MateiBarbu Matei Barbu_Matei Data 9 ianuarie 2025 18:37:52
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.96 kb
#include <bits/stdc++.h>
using namespace std;

const int p = 3;
string text, word;
unordered_map<unsigned int, bool> m;

unsigned int hashF(string word) {
    unsigned int res = 0;
    for (int i = 0; i < word.size(); ++i) {
        res = res * p + (word[i] - 'a');
    }
    return res;
}

int calcPow(int base, int power) {
    int res = 1;
    for (int i = 1; i <= power; ++i) {
        res *= base;
    }
    return res;
}

int main() {
    ifstream cin("abc2.in");
    ofstream cout("abc2.out");
    cin >> text;
    while (cin >> word) {
        m[hashF(word)] = true;
    }
    int wSize = word.size();
    unsigned int res = 0, pos = 0, ans = 0;
    unsigned int last = calcPow(3, wSize - 1);
    for (int i = 0; i < text.size(); ++i) {
        res = res * p + (text[i] - 'a');
        if (i >= wSize - 1) {
            ans += m[res];
            res = res - last * (text[i - (wSize - 1)] - 'a');
        }
    }
    cout << ans;
}