Cod sursa(job #2634253)

Utilizator stefantagaTaga Stefan stefantaga Data 10 iulie 2020 11:05:23
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 1.58 kb
#include <bits/stdc++.h>
#define MOD 666013
#define baza 5
#define MOD2 1000003
#define baza2 7
using namespace std;
ifstream f("abc2.in");
ofstream g("abc2.out");
long long put(long long a,long long b)
{
    if (b==0)
    {
        return 1;
    }
    long long rez=put(a,b/2);
    if (b%2==0)
    {
        return (rez*rez)%MOD;
    }
    return ((rez*rez)%MOD*a)%MOD;
}
long long put2(long long a,long long b)
{
    if (b==0)
    {
        return 1;
    }
    long long rez=put2(a,b/2);
    if (b%2==0)
    {
        return (rez*rez)%MOD2;
    }
    return ((rez*rez)%MOD2*a)%MOD2;
}
long long inv (long long a)
{
    return put(a,MOD+2);
}
char s[10000001],s1[25];
long long n1,nr,n,imp,val1,nr1,lung,j,sum,q,val2,p;
long long fr[50005];
vector <long long > v[666015];
bool ok (long long a)
{
    int j=0,rest=a%MOD;
    for (j=0;j<v[rest].size();j++)
    {
        if (v[rest][j]==a)
        {
            return 1;
        }
    }
    return 0;
}
int main()
{
    f>>s;
    n=strlen(s);
    q=0;
    while (f>>s1)
    {
        n1=strlen(s1);
        nr=0;
        for (j=0;j<n1;j++)
        {
            nr=nr*3+(s1[j]-'a');
        }
        int rest=nr%MOD;
        v[rest].push_back(nr);
    }
    nr=0;
    p=1;
    for (j=1;j<=n1-1;j++)
    {
        p=p*3;
    }
    nr=0;
    for (j=0;j<n1;j++)
    {
        nr=nr*3+(s[j]-'a');
    }
    sum=sum+ok(nr);
    for (j=1;j<n-n1+1;j++)
    {
        nr=nr-p*(s[j-1]-'a');
        nr=nr*3+(s[j+n1-1]-'a');
        sum=sum+ok(nr);
    }
    g<<sum;
    return 0;
}