Cod sursa(job #2174502)

Utilizator dragomirmanuelDragomir Manuel dragomirmanuel Data 16 martie 2018 12:19:24
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.04 kb
#include <iostream>
#include <cstring>
#include <cstdio>

using namespace std;

const int NMax = 2000005;

int pi[NMax];
char a[NMax], b[NMax];
int na, nb;

void Prefix()
{
    int j=0;

    for(int i=1; i<na; ++i)
    {
        while(j>0 && a[i]!=a[j])
            j = pi[j-1];

        if(a[i] == a[j])
            ++j;

        pi[i] = j;
    }
}

void KMP()
{
    int potr = 0;
    int sol[1005];
    int j = 0;

    for(int i=0; i<nb; ++i)
    {
        while(j>0 && b[i] != a[j])
            j = pi[j-1];

        if(b[i] == a[j])
            ++j;

        if(j==na)
        {
            ++potr;
            if(potr<=1000)
                sol[potr] = i - na + 1;
        }
    }

    cout << potr << "\n";
    for(int i=1; i<=min(potr,1000); ++i)
        cout << sol[i] << " ";

}

int main()
{
    freopen("strmatch.in", "r", stdin);
    freopen("strmatch.out", "w", stdout);

    scanf("%s\n%s", &a, &b);

    na = strlen(a);
    nb = strlen(b);

    Prefix();
    KMP();
    return 0;
}