Cod sursa(job #2807281)

Utilizator andrei_C1Andrei Chertes andrei_C1 Data 23 noiembrie 2021 16:59:38
Problema Potrivirea sirurilor Scor 40
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.31 kb
#include <fstream>
#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 ans_size;
std :: vector<int> ans;

void z_function() {
    z[0] = 0;
//    for(int i = 1; i < M; i++) {
//        while(i + z[i] < M && B[i + z[i]] == B[z[i]]) {
//            z[i]++;
//        }
//    }

    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(z[i] == N) {
            if(++ans_size <= 1000) {
                ans.push_back(i - N - 1);
            }
        }
    }
}

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

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

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

    B = A + "#" + B;
    M = B.size();

    z_function();

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

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