Cod sursa(job #1118742)

Utilizator visanrVisan Radu visanr Data 24 februarie 2014 13:00:38
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.01 kb
#include <cstdio>
#include <cstring>
#include <vector>
using namespace std;

const int NMAX = 2000010;

int N, M, PI[NMAX], MatchNr;
char A[NMAX], B[NMAX];
vector<int> MatchPos;

void Build_PI()
{
    int Q = 0;
    PI[1] = 0;
    for(int i = 2; i <= N; ++ i)
    {
        while(Q && A[Q + 1] != A[i]) Q = PI[Q];
        if(A[Q + 1] == A[i]) Q ++;
        PI[i] = Q;
    }
}

void Find_Matches()
{
    int Q = 0;
    for(int i = 1; i <= M; ++ i)
    {
        while(Q && A[Q + 1] != B[i]) Q = PI[Q];
        if(A[Q + 1] == B[i]) Q ++;
        if(Q == N)
        {
            MatchNr ++;
            if(MatchNr <= 1000) MatchPos.push_back(i - N);
        }
    }
}

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

    gets(A + 1); N = strlen(A + 1);
    gets(B + 1); M = strlen(B + 1);

    Build_PI();
    Find_Matches();

    printf("%i\n", MatchNr);
    for(int i = 0; i < MatchPos.size(); ++ i) printf("%i ", MatchPos[i]);
}