Cod sursa(job #1949874)

Utilizator ticozaurStratila Andrei ticozaur Data 2 aprilie 2017 14:47:32
Problema Potrivirea sirurilor Scor 14
Compilator cpp Status done
Runda Arhiva educationala Marime 1.17 kb
#include <iostream>
#include <fstream>
#include <cstring>
#define LMAX 2000000
using namespace std;
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
char a[LMAX],b[LMAX];
int pref[LMAX],n,m,sol[1001],nr;
void prefix()
{
    int j=0,i=1;
    while(i<n)
    {
        if(a[j]==a[i])
        {
            pref[i]=j+1;
            i++;
            j++;
        }
        else
        {
            if(pref[i-1]!=0 && j>0)
                j=pref[j-1];
            else
                i++;
        }
    }
}
void kmp()
{
    int i=0,j=0;
    while(i<m)
    {

        if(b[i]==a[j])
        {
            if(j==n-1){
                nr++;
                if(nr<=1000)
                    sol[nr]=i-j;
                j=-1;
                i--;
            }
            i++;
            j++;
        }
        else
        {
            i++;
            if(j>1)
                j=pref[j-1];
        }
    }
}
int main()
{
    fin>>a;
    fin>>b;
    n=strlen(a);
    m=strlen(b);
    prefix();
    kmp();
    fout<<nr<<'\n';
    nr=min(nr,1000);
    for(int i=1; i<=nr; i++)
        fout<<sol[i]<<' ';
    return 0;
}