Cod sursa(job #1671130)

Utilizator robertkarolRobert Szarvas robertkarol Data 1 aprilie 2016 13:02:24
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 kb
#include <iostream>
#include <fstream>
#include <cstring>
#include <vector>
using namespace std;
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
char a[2000001],b[2000001];
int pi[2000001],m,n,i,v[2000001],nr;
void prefix()
{
    int i,k=0;
    for(i=2;i<=m;i++)
    {
        while(k&&a[k+1]!=a[i]) k=pi[k];
        if(a[k+1]==a[i]) k++;
        pi[i]=k;
    }
}
void kmp()
{
    int i,k=0;
    prefix();
    for(i=1;i<=n;i++)
    {
        while(k&&a[k+1]!=b[i]) k=pi[k];
        if(a[k+1]==b[i]) k++;
        if(k==m)
        {
            k=pi[m];
            nr++;
            if(nr<=1000) v[nr]=i-m;
        }
    }
}
int main()
{
    fin>>a>>b;
    n=strlen(b); m=strlen(a);
    for(i=m;i>=1;i--) a[i]=a[i-1];
    for(i=n;i>=1;i--) b[i]=b[i-1];
    b[0]=a[0]=' ';
    kmp();
    fout<<nr<<"\n";
    n=min(nr,1000);
    for(i=1;i<=n;i++)
        fout<<v[i]<<" ";
    return 0;
}