Cod sursa(job #2292011)

Utilizator ValentinSavoiuFMI Savoiu Valentin-Marian ValentinSavoiu Data 28 noiembrie 2018 21:13:56
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.12 kb
#include <fstream>
#include <string>
#include <vector>
#define MOD 2017
using namespace std;
ifstream f("abc2.in");
ofstream g("abc2.out");
string s1, s2;
int N, M, i ,j, sol;
long long cod, p;
vector <int> Map[MOD];
bool isThere(long long x) {
    int key = x % MOD;
    for (auto it = Map[key].begin(); it != Map[key].end(); ++it)
        if ( *it == x ) return 1;
    return 0;
}
void Push(long long x){
    int key = x % MOD;
    if (! isThere(x))
        Map[key].push_back(x);
}
int main() {
    f >> s1;
    f >> s2;
    N = s1.length();
    M = s2.length();
    cod = 0;
    for (i = 0; i < M; ++i) {
        cod = cod * 3 + (s2[i] - 'a') ;
    }
    Push(cod) ;
    while (f >> s2) {
        cod = 0;
        for (i = 0; i < M; ++i) {
            cod = cod * 3 + (s2[i] - 'a');
        }
        Push(cod);
    }
    cod = 0;
    p = 1;
    for (i = 0; i < M; ++i) {
        cod = cod * 3 + (s1[i] - 'a');
        p = p * 3;
    }
    if (isThere(cod)) ++sol;
    p /= 3;
    for (i = M; i < N; ++i) {
        cod = (cod - (s1[i - M] - 'a' ) * p ) * 3 + (s1[i] - 'a');
        if ( isThere(cod)) ++sol;
    }
    g << sol;
    return 0;
}