Cod sursa(job #2056342)

Utilizator george.ursachiUrsachi George george.ursachi Data 4 noiembrie 2017 11:08:13
Problema Potrivirea sirurilor Scor 14
Compilator cpp Status done
Runda Arhiva educationala Marime 0.94 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,105);
    f.getline(B,105);
    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;
}