Cod sursa(job #2474425)

Utilizator alex_bb8Banilean Alexandru-Ioan alex_bb8 Data 15 octombrie 2019 11:01:07
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.23 kb
#include <bits/stdc++.h>

using namespace std;

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

const int mod=766013,mod1=19017;
long long p3[35],sol;
int n;
char v[10000205],s[35];

vector<long long> M[mod],N[mod1];

void inser(long long x)
{
  int poz=x%mod;
  bool ok=1;
  for(int i=0;i<M[poz].size();i++)
   if(M[poz][i]==x) ok=0;
  if(ok==1) M[poz].push_back(x);
  ok=1;

  int poz1=x%mod1;

  for(int i=0;i<N[poz1].size();i++)
   if(N[poz1][i]==x) ok=0;
  if(ok==1) N[poz1].push_back(x);
}


int query(long long x)
{
  int poz=x%mod,poz1=x%mod1;
  int u=M[poz].size();
  int u1=N[poz1].size();
  if(u==u1 && u!=0) return 1;
}


int main()
{
    p3[0]=1;
    for(int i=1;i<=21;i++)
     p3[i]=p3[i-1]*3;

    f>>v;
    long long nr=0;

    while(f>>s)
    {
     for(int i=0;s[i];i++)
      nr+=(s[i]-'a')*p3[i];

     inser(nr);
     //g<<nr<<"\n";
     nr=0;
     n=strlen(s);
    }

    //g<<"\n";
    nr=0;

    for(int j=0;j<=n-1;j++)
     nr+=(v[j]-'a')*p3[j];

    int upl=strlen(v);

    for(int j=n;j<upl;j++)
    {
     sol+=query(nr);
     //g<<query(nr)<<" ";
     //g<<nr<<"\n";
     nr=nr/3;
     nr+=(v[j]-'a')*p3[n-1];

    }

    g<<sol;
    return 0;
}