Cod sursa(job #2296816)

Utilizator mirunazMiruna Zavelca mirunaz Data 5 decembrie 2018 01:03:08
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.1 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, bool> h;
    char sir[M], s[22];
    in.getline(sir, M);

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

    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] == false) {
            ct ++;
        }
    }
    out << ct;
    return 0;
}