Cod sursa(job #2768649)

Utilizator DragosC1Dragos DragosC1 Data 11 august 2021 17:58:48
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.1 kb
#include <fstream>
#include <algorithm>
#include <iostream>
#include <unordered_map>
using namespace std;
 
string s, aux;
unsigned int base;
int l, l2;
 
unordered_map<unsigned int, bool> mp;
 
void read() {
    int i, j;
    unsigned int hash;
    ifstream f("abc2.in");
    f >> s;
    while (f >> aux) {
        hash = 0;
        base = 1;
        for (j = 0; j < aux.size(); j++) 
            hash = hash * 3 + (aux[j] - 'a');
        l2 = aux.size();
        if (mp.find(hash) == mp.end()) mp[hash] = 1;
    }
}
 
int nr;
 
void solve() {
    int i;
 
    unsigned int hash = 0;
    for (i = 0; i < l2; i++) {
        hash = hash * 3 + (s[i] - 'a');
        base = base * 3;
    }
    base /= 3;
    if (mp.find(hash) != mp.end())
        nr++;
 
    for (i = l2; i < s.size(); i++) {
        hash = (hash - (s[i - l2] - 'a') * base) * 3 + (s[i] - 'a');
        if (mp.find(hash) != mp.end())
            nr++;
    }
}
 
void output() {
    ofstream g("abc2.out");
    g << nr;
    g.close();
}
 
int main() {
    read();
    solve();
    output();
    return 0;
}