Cod sursa(job #2976774)

Utilizator indianu_talpa_iuteTisca Catalin indianu_talpa_iute Data 9 februarie 2023 23:15:36
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.89 kb
#include <bits/stdc++.h>
#define MAXSZ 2000005

using namespace std;

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

int n, m, lps[MAXSZ];
char a[MAXSZ + 1], b[MAXSZ + 1];

int main() {
    fin.getline(a, MAXSZ);
    fin.getline(b, MAXSZ);
    n = (int) strlen(a);
    m = (int) strlen(b);

    int q = 0;
    for (int i = 2; i <= n; i++) {
        while (q && a[q] != a[i - 1])
            q = lps[q];
        if (a[q] == a[i - 1])
            q++;
        lps[i] = q;
    }

    vector<int> matches;
    q = 0;
    for (int i = 0; i < m; i++) {
        while (q && a[q] != b[i])
            q = lps[q];
        if (a[q] == b[i])
            q++;
        if (q == n) {
            matches.push_back(i - n + 1);
            q = lps[q];
        }
    }

    fout << matches.size() << '\n';
    for (int i = 0; i < min((int) matches.size(), 1000); i++)
        fout << matches[i] << ' ';
    return 0;
}