Cod sursa(job #2398380)

Utilizator mihaimodiMihai Modi mihaimodi Data 5 aprilie 2019 13:30:35
Problema Abc2 Scor 0
Compilator cpp-64 Status done
Runda Arhiva de probleme Marime 0.99 kb
#include <fstream>
#include <cstring>
#include <vector>
#define mod 10007
using namespace std;
ifstream fin ("abc2.in ");
ofstream fout("abc2.out");
vector <int> v[10008];
char s1[10000005],s2[22];
int p[22],sol;
int caut(int x,int k)
{
    for(int i=0;i<v[k].size();i++)
        if(v[k][i]==x)
            return 1;
    return 0;
}
int main()
{
    fin>>(s1+1);
    p[0]=1;
    for(int i=1;i<=19;i++)
        p[i]=p[i-1]*3;
    while(fin>>(s2+1))
    {
        int x=0;
        for(int i=1;s2[i]!=0;i++)
            x=x*3+(s2[i]-'a');
    int   k=x%mod;
        if(!caut(x,k))
            v[k].push_back(x);
    }
    int n=strlen(s1+1);
    int m=strlen(s2+1);
    int nr=0;
    sol=0;
    for(int i=1;i<=m;i++)
        nr=nr*3+(s1[i]-'a');
    if(caut(nr,nr%mod))
        sol=1;

    for(int i=m+1;s1[i]!=0;i++)
    {
        nr=(nr-p[m-1]*(s1[i-m]-'a'))*3+(s1[i]-'a');
        if(caut(nr,nr%mod)==1)
            sol++;
    }
    fout<<sol;
    return 0;
}