Cod sursa(job #2102445)

Utilizator moise_alexandruMoise Alexandru moise_alexandru Data 8 ianuarie 2018 20:31:09
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.14 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;
ifstream in("abc2.in");
ofstream out("abc2.out");
const int mod = 30000;
vector <long long> hsh[mod + 5];
string text;

inline void push(long long x)
{
    int p = x % mod;
    hsh[p].push_back(x);
}

inline bool searchfor(long long x)
{
    int p = x % mod;
    for(auto it : hsh[p])
        if(it == x)
            return 1;
    return 0;
}

int main()
{
    cin.sync_with_stdio(0);
    in >> text;
    string cuv;
    while(in >> cuv)
    {
        long long p = 0;
        for(auto it : cuv)
            p = p * 3 + it - 'a';
        push(p);
    }
    int lg = cuv.size();
    long long put3 = 1;
    long long a = 0;
    for(int i = 0; i < lg; i++)
    {
        a = a * 3 + text[i] - 'a';
        if(i != 0)
            put3 = put3 * 3;
    }
    int nr = 0;
    for(int i = lg; i < text.size(); i++)
    {
        if(searchfor(a) == 1)
            nr++;
        a = a - put3 * (text[i - lg] - 'a');
        a = a * 3 + text[i] - 'a';
    }
    if(searchfor(a) == 1)
        nr++;
    out << nr << "\n";
    return 0;
}