Cod sursa(job #1105733)

Utilizator gabrielinelusGabriel-Robert Inelus gabrielinelus Data 12 februarie 2014 00:12:48
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.12 kb
#include <cstdio>
#include <cstring>
#include <vector>

#define Lmax 2000005

using namespace std;

char A[Lmax],B[Lmax];
int P[Lmax],N,M;
vector<int> v;

void make_prefix()
{
    int i , q = 0;
    for(int i = 2; i <= N; ++i)
    {
        while(q && A[i] != A[q+1])
            q = P[q];
        if(A[i] == A[q+1])
            ++q;
        P[i] = q;
    }
}

void make_pattern()
{
    int i , q = 0 ,nre = 0;
    for(i = 1; i <= M; ++i)
    {
        while(q && B[i] != A[q+1])
            q = P[q];
        if(B[i] == A[q+1])
            ++q;
        if(q == N)
        {
            q = P[q];
            if(++nre <= 1000)
                v.push_back(i-N);
        }
    }
}

void KMP()
{
    make_prefix();
    make_pattern();
}

void print()
{
    printf("%d\n",v.size());
    for(int i = 0; i < (int)v.size(); ++i)
        printf("%d ",v[i]);
}

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

    scanf("%s%s",A+1,B+1);
    A[0] = B[0] = '*';
    N = strlen(A) - 1,M = strlen(B) - 1;

    KMP();
    print();

    return 0;
}