Cod sursa(job #2474976)

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

using namespace std;

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

const int mod=14029;
vector<unsigned int> M[mod];
bool okmod[mod];
string v,s;
unsigned int p3[35],sol=0;
unsigned int n,poz,i,j;

int main()
{




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

    f>>v;

    long long nr=0;

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

    unsigned int poz=nr%mod;
    bool ok=1;
    for(unsigned int it : M[poz])
     if(it==nr) ok=0;
    if(ok==1) M[poz].push_back(nr);
     nr=0;
     n=s.size();
    }


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

    unsigned int upl=v.size();

    for(j=n;j<upl;j++)
    {
     poz=nr%mod;
     if(!okmod[poz]) sort(M[poz].begin(),M[poz].end()),okmod[poz]=1;
     if(binary_search(M[poz].begin(),M[poz].end(),nr))
      sol++;
     nr=nr/3;
     nr+=(v[j]-'a')*p3[n-1];
    }

     poz=nr%mod;
     if(!okmod[poz]) sort(M[poz].begin(),M[poz].end()),okmod[poz]=1;
     if(binary_search(M[poz].begin(),M[poz].end(),nr))
      sol++;


    g<<sol;
    return 0;
}