Cod sursa(job #2203167)

Utilizator YouDontNeedMyNameJurcut Paul YouDontNeedMyName Data 11 mai 2018 11:24:33
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.08 kb
#include <bits/stdc++.h>
#define N_MAX 2000005
using namespace std;
ifstream in("strmatch.in");
ofstream out("strmatch.out");
int n,m,lps[N_MAX];
char a[N_MAX],b[N_MAX];
vector <int> sol;
void prelucrare()
{
    int k=0;
    for(int i=1; i<n; i++)
    {
        if(a[i]==a[k])
        {
            k++;
        }
        else
        {
            k=0;
            if(a[i]==a[k])
                k++;
        }
        lps[i]=k;
    }
}
void kmp()
{
    int k=0;
    for(int i=0; i<m; i++)
    {
        while( k != 0 && (a[k] != b[i]) )
            k = lps[max(k - 1, 0)];
        if(a[k]==b[i])
        {
            k++;
        }
        if(k==n)
        {
            if(sol.size()<1000)
                sol.push_back(i-k+1);
            else
                return;
            k=lps[k-1];
        }
    }
}
int main()
{
    in >> a;
    n=strlen(a);
    in >> b;
    m=strlen(b);
    prelucrare();
    kmp();
    out << sol.size() << '\n';
    for(int i=0; i<sol.size(); i++)
    {
        out << sol[i] << ' ';
    }
    return 0;
}