Cod sursa(job #2208616)

Utilizator lucaperjuLuca Perju Verzotti lucaperju Data 30 mai 2018 18:05:30
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.56 kb
#include <fstream>
#include <cstring>
using namespace std;
ifstream in ("strmatch.in");
ofstream out ("strmatch.out");
const long long baza=100;
const long long p=961748941;
char a[2000003],b[2000003];
int poz[2000003];
long long put=1,rp=0,n,m,cnt=0;;
void citire ()
{
    in.getline(a,2000000);
    in>>ws;
    in.getline(b,2000000);
    m=strlen(a);
    n=strlen(b);
    for(int i=0;i<m;++i)
    {
        if(a[i]>='0' && a[i]<='9')
            a[i]=a[i]-'0';
        if(a[i]>='a' && a[i]<='z')
            a[i]=a[i]-'a'+10;
        if(a[i]>='A' && a[i]<='Z')
            a[i]=a[i]-'A'+36;
    }
    for(int i=0;i<n;++i)
    {
        if(b[i]>='0' && b[i]<='9')
            b[i]=b[i]-'0';
        if(b[i]>='a' && b[i]<='z')
            b[i]=b[i]-'a'+10;
        if(b[i]>='A' && b[i]<='Z')
            b[i]=b[i]-'A'+36;
    }
}
void frmput ()
{
    for(int i=0;i<m;++i)
        put=(put*1LL*baza)%p;
}
void frmrp ()
{
    for(int i=0;i<m;++i)
    {
        rp=(rp*1LL*baza)%p;
        rp=(rp+a[i])%p;
    }
}
void rez ()
{
    long long i,rc=0;
    for(i=0;i<n;++i)
    {
        rc=(rc*1LL*baza)%p;
        rc=(rc+b[i])%p;
        if(i>=m)
        {
            rc=(rc+p-(put*1LL*b[i-m])%p)%p;
            if(rc==rp)
                poz[++cnt]=i-m+1;
        }
        if(rc==rp && i==m-1)
            poz[++cnt]=i-m+1;
    }
}
int main()
{
    citire();
    frmput();
    frmrp();
    rez();
    out<<cnt<<'\n';
    if(cnt>=1000)
        cnt=1000;
    for(int i=1;i<=cnt;++i)
        out<<poz[i]<<' ';
    return 0;
}