Cod sursa(job #2720244)

Utilizator Uriesu_IuliusUriesu Iulius Uriesu_Iulius Data 10 martie 2021 17:58:48
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.27 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("strmatch.in");
ofstream fout("strmatch.out");

char pat[2000001], txt[2000001];
int lps[2000001];
int n, m;
vector<int> ans;

void Preprocesare()
{
    int i=1, len=0;
    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()
{
    int i=0, j=0;
    while(i<n)
    {
        if(pat[j]==txt[i])
        {
            j++;
            i++;
        }
        if(j==m)
        {
            ans.push_back(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()
{
    fin >> pat;
    fin >> txt;
    m=strlen(pat);
    n=strlen(txt);
    Preprocesare();
    KMP();
    fout << ans.size() << "\n";
    if(ans.size()<=1000)
        for(int i=0; i<ans.size(); i++)
            fout << ans[i] << " ";
    else
        for(int i=0; i<1000; i++)
            fout << ans[i] << " ";
    return 0;
}