Cod sursa(job #2296812)

Utilizator mirunazMiruna Zavelca mirunaz Data 5 decembrie 2018 00:57:30
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.04 kb
#include<fstream>
#include<iostream>
#include<cstring>
#include<unordered_map>
#define M 10000010
using namespace std;

int CalcMask(char s[], int n)
{
    int p = 1, sum = 0;
    for(int i=n - 1; i>=0; i--) {
        sum += (s[i] - 'a')*p;
        p *= 3;
    }
    return sum;
}

int RenewMask(char s[], int n, int start, int pk, int x)
{
    int aux = (s[start+n] - 'a') - (s[start] - 'a')*pk;
    return (aux + 3*x);
}

int main ()
{
    ifstream in("abc2.in");
    ofstream out("abc2.out");
    unordered_map <int, int> h;
    char sir[M], s[22];
    in.getline(sir, M);

    int n, pk=1;
    while(in.getline(s, 22)) {
        n = strlen(s);
        h[CalcMask(s, n)] ++;
    }

    for (int i=0;i<n;++i){
        pk*=3;
    }

    int x = CalcMask(sir, n);
    int ct = 0;
    if(h[x] != 0) {
        ct ++;
    }

    int m = strlen(sir);
    for(int i=0; i<m-n; i++) {
        x = RenewMask(sir, n, i, pk, x);
        if(h[x] != 0) {
            ct ++;
        }
    }
    out << ct;
    return 0;
}