Cod sursa(job #2738595)

Utilizator vlad082002Ciocoiu Vlad vlad082002 Data 6 aprilie 2021 09:16:50
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.8 kb
#include <bits/stdc++.h>
using namespace std;

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

string a, b;

int main() {
    fin >> a >> b;

    vector<int> pi(a.length()+5, 0);
    for(int i = 1; i < a.length(); i++) {
        pi[i] = pi[i-1];
        while(pi[i] > 0 && a[i] != a[pi[i]])
            pi[i] = pi[pi[i]-1];
        if(a[i] == a[pi[i]]) pi[i]++;
    }

    int q = 0;
    int nr = 0;
    vector<int> ans;
    for(int i = 0; i < b.length(); i++) {
        while(q > 0 && a[q] != b[i])
            q = pi[q-1];
        if(a[q] == b[i]) q++;
        if(q == a.length()) {
            nr++;
            if(ans.size() < 1000)
                ans.push_back(i-a.length()+1);
        }
    }

    fout << nr << '\n';
    for(auto x: ans) fout << x << ' ';
}