Cod sursa(job #2739086)

Utilizator BuzatuCalinBuzatu Calin BuzatuCalin Data 6 aprilie 2021 19:58:38
Problema Potrivirea sirurilor Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.34 kb
//problema KMP
#include <iostream>
#include <fstream>
using namespace std;
#define DIM 1000001
#define DIM2 1001
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
int lps[DIM],indici[DIM2],nr_indici;
void prelucrareLPS(string pat,int m)
{
    int len=0,i=1;
    lps[0]=0;
    while(i<m)
    {
        if(pat[i]==pat[len])
        {
            len++;
            lps[i]=len;
            i++;
        }
        else
        {
            if(len!=0)
            {
                len=lps[len-1];
            }
            else
            {
                lps[i]=0;
                i++;
            }
        }
    }
}
void KMP(string pat,string txt)
{
    int n=txt.length(),m=pat.length(),i=0,j=0;
    prelucrareLPS(pat,m);
    while(i<n)
    {
        if(pat[j]==txt[i])
        {
            i++;j++;
        }
        if(j==m)
        {
            indici[nr_indici++]=i-j;
            j=lps[j-1];
        }
        else if(i<n && pat[j]!=txt[i])
        {
            if(j!=0)
            {
                j=lps[j-1];
            }
            else
            {
                i++;
            }
        }
    }
}
int main()
{
    string A,B;
    fin>>A>>B;
    KMP(A,B);
    fout<<nr_indici<<'\n';
    for(int i=0;i<nr_indici && i<1000;i++)
    {
        fout<<indici[i]<<" ";
    }
}