Cod sursa(job #2261188)

Utilizator SqueekDanielTodasca Daniel SqueekDaniel Data 16 octombrie 2018 01:26:06
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.15 kb
#include <bits/stdc++.h>

#define Petrisor void
#define MaxN 2000005
#define MaxA 1000

std::ifstream InFile("strmatch.in");
std::ofstream OutFile("strmatch.out");

char A[MaxN], B[MaxN];
int M, N, NAns, LPS[MaxN];
std::vector <int> Ans;

Petrisor PrecomputeLPS() {
    int Len = 0;
    for (int i=2; i<=N; ++i) {
        while (Len && A[i] != A[Len+1])
            Len = LPS[Len];
        if (A[Len+1] == A[i])
            ++Len;

        LPS[i] = Len;
    }
}

Petrisor KMP() {
    PrecomputeLPS();

    int Len = 0;
    for (int i=1; i<=M; ++i) {
        while(Len && B[i] != A[Len+1])
            Len = LPS[Len];
        if (A[Len+1] == B[i])
            ++Len;

        if (Len == N) {
            NAns ++;
            if (NAns <= MaxA)
                Ans.push_back(i-N);
        }
    }
}

Petrisor Citire() {
    InFile >> A+1 >> B+1;
    N = strlen(A+1);
    M = strlen(B+1);
}

Petrisor Rezolvare() {
    KMP();

    OutFile << NAns << '\n';
    for (int i=0; i<Ans.size(); ++i)
        OutFile << Ans[i] << ' ' ;
    OutFile << '\n';
}

int main()
{
    Citire();
    Rezolvare();

    return 0;
}