Cod sursa(job #2474417)

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

using namespace std;

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

const int dim1 = 666013, dim2 = 10000001, dim3 = 21;

vector <int> v0[dim1];

long long nrgas;
long long p3[dim3];
char v[dim2], cuv[dim3];

int main()
{
    int i, j, l, nr = 0, rest;

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

    f >> v;

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

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

    for(i = l; v[i]; ++i){
        rest = nr % dim1;
        v0[rest].push_back(nr);

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

    rest = nr % dim1;
    v0[rest].push_back(nr);

    do{
        nr = 0;

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

        rest = nr % dim1;
        for(i = 0; i < v0[rest].size(); ++i){
            if(v0[rest].at(i) == nr){
                ++nrgas;
                break;
            }
        }

    }while(f >> cuv);

    g << nrgas;

    return 0;
}