Cod sursa(job #2821877)

Utilizator ana_madalina_18Radu Ana Madalina ana_madalina_18 Data 23 decembrie 2021 10:18:53
Problema Potrivirea sirurilor Scor 14
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.89 kb
#include <fstream>
#include <cstring>
#include <vector>
#include <algorithm>
using namespace std;
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
string a, b;
vector <int> v;
int base=97;
int mod=666019;
int base2=89;
int mod2=714199;
long long pow(int a,int putere,int Mod)
{
    if(putere==1)
    {
        return a;
    }
    else if(putere%2==0)
    {
        long long p=pow(a,putere/2,Mod);
        return p*p%Mod;
    }
    else
    {
        long long p=pow(a,putere/2,Mod);
        return a*p*p%Mod;
    }
}
long long Hash(int st, int dr, const string & x,int Base,int Mod)
{
    long long rasp=0;
    long long p=1;
    for(int i=st;i<=dr;i++)
    {
        rasp=rasp+x[i]*p;
        rasp=rasp%Mod;
        p=p*Base;
        p=p%Mod;
    }
    return rasp;
}
int main()
{
    fin>>a;
    fin>>b;
    int lung_a=a.size();
    int lung_b=b.size();
    long long h_a=Hash(0,lung_a-1,a,97,666019);
    long long h_a2=Hash(0,lung_a-1,a,89,714199);

    long long h_b=Hash(lung_b-lung_a,lung_b-1,b,97,666019);
    long long h_b2=Hash(lung_b-lung_a,lung_b-1,b,89,714199);

    long long putere=pow(base,lung_a-1,mod);
    long long putere2=pow(base2,lung_a-1,mod2);

    int ult=lung_b-1;
    for(int i=lung_b-lung_a;i>=0;i--)
    {
        if(h_b==h_a && h_b2==h_a2)
        {
            v.push_back(i);
        }
        if(i!=0)
        {
            h_b=(h_b-b[ult]*putere+mod)%mod;
            h_b*=base;
            h_b+=b[i-1];
            h_b=h_b%mod;

            h_b2=(h_b2-b[ult]*putere2+mod2)%mod2;
            h_b2*=base2;
            h_b2+=b[i-1];
            h_b2=h_b2%mod2;

            ult--;
        }
    }
    int lung_v=v.size();
    fout<<lung_v<<'\n';
    //reverse(v.begin(),v.end());
    int dist=min(1000,lung_v);
    while(dist!=0)
    {
        dist--;
        fout<<v[dist]<<' ';
    }
    return 0;
}