Cod sursa(job #2252899)

Utilizator YouDontNeedMyNameJurcut Paul YouDontNeedMyName Data 3 octombrie 2018 11:42:32
Problema Potrivirea sirurilor Scor 38
Compilator cpp Status done
Runda Arhiva educationala Marime 1.21 kb
#include <bits/stdc++.h>
#define Lmax 2000005
using namespace std;
ifstream in("strmatch.in");
ofstream out("strmatch.out");
int n,lcs[Lmax],m,nr,s[Lmax];
char a[Lmax],b[Lmax];
void calc_lcs()
{
    int i=1,len=0;
    while(i<n)
    {
        if(a[i]==a[len])
        {
            len++;
            lcs[i]=len;
            i++;
        }
        else
        {
            if(len>0)
            {
                len=lcs[len-1];
            }
            else
            {
                lcs[i]=0;
                i++;
            }
        }
    }
}
void solve()
{
    for(int i=0,j=0; i<m;)
    {
        if(j==n)
        {
            s[++nr]=i;
            if(nr>1001)
                return;
            j=lcs[j-1];
        }
        else if(a[j]==b[i])
        {
            i++;
            j++;
        }
        else
        {
            if(j)
                j=lcs[j-1];
            else
                i++;
        }
    }
}
int main()
{
    in >> a;
    n=strlen(a);
    in >> b;
    m=strlen(b);
    calc_lcs();
    solve();
    out << nr << '\n';
    for(int i=1; i<=min(nr,1000); i++)
    {
        out << s[i]-n << ' ';
    }
    return 0;
}