Cod sursa(job #2417322)

Utilizator usureluflorianUsurelu Florian-Robert usureluflorian Data 29 aprilie 2019 15:27:31
Problema Potrivirea sirurilor Scor 14
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.19 kb
#include <bits/stdc++.h>
#define mod1 1000000009
#define mod2 666013
#define int long long
using namespace std;
ifstream f ("strmatch.in");
ofstream g ("strmatch.out");
const int nmax=2e6+3;
int pwp1,pwp2,cod1,cod2,n,t,usu1,usu2;
char s1[nmax],s[nmax];
vector <int> sol;
main()
{
    f>>(s1+1);
    f>>(s+1);
    n=strlen(s1+1);
    t=strlen(s+1);
    for(int i=1;i<=n;++i)
    {
        cod1=(cod1*26+(s1[i]-'A'))%mod1;
        while(cod1<0) cod1+=mod1;
        cod2=(cod2*26+(s1[i]-'A'))%mod2;
        while(cod2<0) cod2+=mod2;
    }
    pwp1=1ll;
    pwp2=1ll;
    for(int i=1;i<n;++i)
    {
        pwp1=pwp1*26%mod1;
        pwp2=pwp2*26%mod2;
    }
    for(int i=1;i<=t;++i)
    {
        if(i<=n) usu1=(usu1*26+(s[i]-'A'))%mod1;
        else usu1=((usu1-(s[i-n]-'A')*pwp1)*26+(s[i]-'A'))%mod1;
        if(i<=n) usu2=(usu2*26+(s[i]-'A'))%mod2;
        else usu2=((usu2-(s[i-n]-'A')*pwp2)*26+(s[i]-'A'))%mod2;
        while(usu2<0) usu2+=mod1;
        if(i>=n&&usu1==cod1&&usu2==cod2)
        {
            sol.push_back(i-n+1);
            if(sol.size()==1000) break;
        }
    }
    g<<sol.size()<<'\n';
    for(int i=0;i<sol.size();++i) g<<sol[i]-1<<' ';
}