Cod sursa(job #2807324)

Utilizator andrei_C1Andrei Chertes andrei_C1 Data 23 noiembrie 2021 17:40:21
Problema Potrivirea sirurilor Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.26 kb
#include <fstream>
#include <iostream>
#include <algorithm>
#include <string>
#include <vector>

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

const int DIM = 4e6;

int N, M;
std :: string A, B;
int z[DIM + 1];

int cnt;
std :: vector<int> ans;

void z_function() {
    z[0] = 0;

    int right = 0, left = 0;
    for(int i = 1; i < M; i++) {
        if(right >= i) {
            z[i] = std :: min(z[i - left], right - i + 1);
        }

        while(i + z[i] < M && B[i + z[i]] == B[z[i]]) {
            z[i]++;
        }

        if(i + z[i] - 1 > right) {
            right = i - z[i] + 1;
            left = i;
        }

        if(i > N && z[i] == N) {
            cnt++;
            if(ans.size() + 1 <= 1000) {
                ans.push_back(i - N - 1);
            }
        }
    }
}

int main() {
    fin >> A >> B;

    N = A.size();
    M = B.size();

//    std :: cout << N << " " << M << '\n';

    if(N > M) {
        fout << "0\n";
        return 0;
    }

    B = A + "#" + B;

    M = B.size();

    z_function();

    fout << cnt << '\n';
    for(int i = 0; i < ans.size(); i++) {
        fout << ans[i] << " ";
    }

    fin.close();
    fout.close();
    return 0;
}