Cod sursa(job #2056348)

Utilizator george.ursachiUrsachi George george.ursachi Data 4 noiembrie 2017 11:10:06
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.95 kb
#include <iostream>
#include <fstream>
#include <string.h>
using namespace std;

ifstream f("strmatch.in");
ofstream g("strmatch.out");
char A[20000005],B[2000005];
int pos[20000005];
int n,pi[20000005],i,M,N,q;

void prefix()
{
    int i, q=0;
    pi[1]=0;
    for(i=2;i<=M;++i)
    {
        while(q && A[q+1]!=A[i])q=pi[q];
        if(A[q+1]==A[i])++q;
        pi[i]=q;
    }
}

void KMP()
{
    for(i=1;i<=N;i++)
    {
        while(q && A[q+1]!=B[i])q=pi[q];
        if(A[q+1]==B[i])++q;
        if(q==M)
        {
            q=pi[M];
            ++n;
            pos[n]=i-M;
        }
    }
}

int main()
{
    f.getline(A,20000005);
    f.getline(B,20000005);
    M=strlen(A);
    N=strlen(B);

    for(i=M;i;--i)A[i]=A[i-1];
    A[0]=' ';

    for(i=N;i;--i)B[i]=B[i-1];
    B[0]=' ';

    prefix();
    KMP();

    g<<n<<'\n';
    if(n>1000) n=1000;

    for(i=1;i<=n;i++) g<<pos[i]<<' ';

    return 0;
}