Cod sursa(job #2387394)

Utilizator YouDontNeedMyNameJurcut Paul YouDontNeedMyName Data 24 martie 2019 16:50:58
Problema Potrivirea sirurilor Scor 38
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.21 kb
#include <bits/stdc++.h>
#define mod 698365723
#define mod2 530957673
#define p 21564321
using namespace std;
ifstream in("strmatch.in");
ofstream out("strmatch.out");
char a[2000005],b[2000005];
long long nr1,nr2,cs1,cs2;
vector <int> sol;
int main()
{
    in >> a;
    in >> b;
    long long pr=1,p2=1;
    for(int i=0; i<strlen(a); i++)
    {
        nr1=(nr1*p+a[i])%mod;
        nr2=(nr2*p+a[i])%mod2;
        cs1=(cs1*p+b[i])%mod;
        cs2=(cs2*p+b[i])%mod2;
        if(i!=0)
        {
            pr=pr*p%mod;
            p2=p2*p%mod2;
        }
    }
    if(nr1==cs1 && nr2==cs2)
    {
        sol.push_back(strlen(a)-1);
    }
    for(int i=strlen(a); i<strlen(b); i++)
    {
        cs1=(cs1-pr*b[i-strlen(a)]%mod)%mod;
        while(cs1<0)
            cs1+=mod;
        cs1=(cs1*p+b[i])%mod;
        cs2=(cs2-p2*b[i-strlen(a)]%mod2)%mod2;
        while(cs2<0)
            cs2+=mod2;
        cs2=(cs2*p+b[i])%mod2;
        if(nr1==cs1 && nr2==cs2)
        {
            sol.push_back(i-strlen(a)+1);
        }
    }
    out << sol.size() << '\n';
    int pol=sol.size();
    for(int i=0; i<min(pol,1000); i++)
    {
        out << sol[i] << ' ';
    }
    return 0;
}