Cod sursa(job #2076096)

Utilizator stefdascalescuStefan Dascalescu stefdascalescu Data 26 noiembrie 2017 10:17:15
Problema Abc2 Scor 0
Compilator cpp Status done
Runda Arhiva de probleme Marime 1.47 kb
#define ross 666013
#include<bits/stdc++.h>
using namespace std;
ifstream f("abc2.in");
ofstream g("abc2.out");
long long nrcur;
char c[10000003];
int len,ln,sol;
int R;
vector<int>rest[ross];
char q[25];
int main()
{
    f>>c;
    R=strlen(c);
    while(f>>q)
    {
        long long p3=1;
        nrcur=0;
        if(ln==0)
            ln=strlen(q);
        for(int j=ln-1;j>=0;--j)
            nrcur+=p3*(q[j]-'a'),p3*=3;
        long long R=nrcur%ross;
        rest[R].push_back(nrcur);
    }
    for(int i=0;i<ross;++i)
        if(!rest[i].empty())
            sort(rest[i].begin(),rest[i].end());
    long long p3=1;
    for(int j=0;j<=R-ln;++j)
    {
        if(j==0){
            for(int k=ln-1;k>=0;--k)
                nrcur+=p3*(c[k]-'a'),p3*=3;
            p3/=3;
        }
        else
        {
            nrcur-=p3*(c[j-1]-'a');
            nrcur*=3;
            nrcur+=(c[j+ln-1]-'a');
        }
        long long R=nrcur%ross;
        if(!rest[R].empty()){
            int b=0;
            int e=rest[R].size()-1;
            while(b<=e)
            {
                int m=(b+e)/2;
                if(rest[R][m]==nrcur)
                {
                    ++sol;
                    break;
                }
                else
                    if(rest[R][m]>nrcur)
                        e=m-1;
                    else
                        b=m+1;
            }
        }
    }
    g<<sol;
    return 0;
}