Cod sursa(job #2474475)

Utilizator CiboAndreiAndrei Cibo CiboAndrei Data 15 octombrie 2019 11:54:06
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.39 kb
#include <bits/stdc++.h>

using namespace std;

ifstream f("abc2.in");
ofstream g("abc2.out");

const int dim1 = 15017, dim2 = 10000002, dim3 = 22;

int main()
{
    vector <unsigned int> v0[dim1];

    unsigned int nrgas, nr = 0;
    unsigned int p3[dim3];
    char v[dim2], cuv[dim3];

    unsigned int i, j, l, rest;
    bool ok;

    p3[0] = 1;
    for(i = 1; i < dim3; ++i)
        p3[i] = p3[i - 1] * 3;

    f >> v;

    f >> cuv;
    l = strlen(cuv);

    do{
        nr = 0;
        ok = 1;

        for(i = 0; cuv[i]; ++i){
            nr += p3[i] * (cuv[i] - 'a');
        }

        rest = nr % dim1;

        for(unsigned int it : v0[rest]){
            if(it == nr){
                ok = 0;
                break;
            }
        }

        if(ok)
            v0[rest].push_back(nr);

    }while(f >> cuv);

    nr = 0;
    for(i = 0; i < l; ++i){
        nr += p3[i] * (v[i] - 'a');
    }

    for(j = l; v[j]; ++j){
        rest = nr % dim1;

        for(unsigned int it : v0[rest]){
            if(it == nr){
                ++nrgas;
                break;
            }
        }

        nr /= 3;
        nr += p3[l - 1] * (v[j] - 'a');
    }

    rest = nr % dim1;
    for(unsigned int it : v0[rest]){
        if(it == nr){
            ++nrgas;
            break;
        }
    }

    g << nrgas;

    return 0;
}