Cod sursa(job #2640991)

Utilizator etienAndrone Stefan etien Data 9 august 2020 15:01:26
Problema Potrivirea sirurilor Scor 80
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.5 kb
#include<bits/stdc++.h>
using namespace std;
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
vector<int>poz;
const int MOD=1e9+7;
const int p1=103;
int cc(char c)
{
    if(c>='a'&&c<='z')
        return c-'a'+1;
    if(c>='A'&&c<='Z')
        return c-'A'+27;
    return c-'0'+53;
}
int hsh(string s)
{
    int mod=0,val=1;
    for(int poz=(int)s.size()-1;poz>=0;poz--)
    {
        mod=(mod+1LL*val*cc(s[poz]))%MOD;
        val=(1LL*val*p1)%MOD;
    }
    return mod;
}
int main()
{
    string a,b;
    fin>>a>>b;
    int nr=0,hsa=hsh(a);
    string s=b.substr(0,a.size());
    int hsb=hsh(s);
    int put=1;
    for(int i=1;i<=(int)a.size();i++)
    {
        put=(1LL*put*p1)%MOD;
        //cout<<put<<" ";
    }
    //cout<<hsa<<"\n";
    for(int i=a.size();i<(int)b.size();i++)
    {
        if(hsa==hsb)
        {
            string s=b.substr(i-a.size(),a.size());
            if(s==a)
            {
                nr++;
                if(nr<=1000)
                    poz.push_back(i);
            }
        }
        //cout<<hsb<<" ";
        hsb=(1LL*hsb*p1+cc(b[i]))%MOD;
        hsb-=(1LL*put*(cc(b[i-a.size()])))%MOD;
        if(hsb<0)
            hsb+=MOD;
    }
    if(hsa==hsb)
    {
        string s=b.substr(b.size()-a.size(),a.size());
        if(s==a)
        {
            nr++;
            if(nr<=1000)
                poz.push_back(b.size());
        }
    }
    fout<<nr<<"\n";
    for(auto p:poz)
        fout<<p-a.size()<<" ";
}