Cod sursa(job #2352743)

Utilizator SqueekDanielTodasca Daniel SqueekDaniel Data 23 februarie 2019 17:24:38
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.04 kb
#include <bits/stdc++.h>

#define MAXN 2000005

int M, N;
char B[MAXN], A[MAXN];

int LPS[MAXN];
void ComputeLPS() {
    int Len = 0;
    for (int i=2; i<=N; ++i) {
        while (Len && A[Len+1] != A[i])
            Len = LPS[Len];
        if (A[Len+1] == A[i])
            ++ Len;
        LPS[i] = Len;
    }
}

int Count;
std::vector <int> Ans;
void KMP() {
    ComputeLPS();
    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) {
            ++ Count;
            if (Count <= 1000)
                Ans.push_back(i-N);
        }
    }
}

std::ifstream In ("strmatch.in");
std::ofstream Out("strmatch.out");

void Citire() {
    In >> A+1 >> B+1;
    M = strlen(B+1);
    N = strlen(A+1);
}

void Rezolvare() {
    KMP();
    Out << Count << '\n';
    for (auto idx:Ans)
        Out << idx << ' ';
}

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

    return 0;
}