Cod sursa(job #2569560)

Utilizator ioana_marinescuMarinescu Ioana ioana_marinescu Data 4 martie 2020 12:35:51
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1 kb
#include <bits/stdc++.h>

using namespace std;

ifstream fin("strmatch.in");
ofstream fout("strmatch.out");

const int MAX_L = 2000000;

char P[MAX_L + 5], T[MAX_L + 5];
int lps[MAX_L + 5];
int cnt;

vector<int> ans;

int main() {
    fin >> (P + 1);
    P[0] = '\0';
    int sizeP = strlen(P + 1);

    lps[0] = lps[1] = 0;
    for (int i = 2; i <= sizeP; i++) {
        int x = lps[i - 1];

        while (x > 0 && P[x + 1] != P[i])
            x = lps[x];

        if (P[x + 1] == P[i])
            lps[i] = x + 1;
        else
            lps[i] = 0;
    }

    fin >> T;
    int sizeT = strlen(T);

    int x = 0;
    for (int i = 0; i < sizeT; i++) {
        while(x > 0 && P[x + 1] != T[i])
            x = lps[x];

        if (P[x + 1] == T[i])
            x++;
        else
            x = 0;

        if (x == sizeP) {
            cnt++;
            ans.push_back(i - sizeP + 1);
        }
    }

    fout << cnt << '\n';
    for (int i = 0; i < min(cnt, 1000); i++)
        fout << ans[i] << ' ';

    return 0;
}