Cod sursa(job #3159829)

Utilizator cristianabalcanuCristiana Balcanu cristianabalcanu Data 22 octombrie 2023 12:06:41
Problema Potrivirea sirurilor Scor 78
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.92 kb
#include <bits/stdc++.h>
#define NN 2000005
using namespace std;
ifstream fin ("strmatch.in");
ofstream fout ("strmatch.out");

int pr[NN];
string a, b;
vector <int> ans;

int main()
{
    fin >> a >> b;
    int k = 0;
    a.push_back('#');
    for(int i = 1 ; i < a.size() ; i++)
    {
        while(k != 0 && a[k] != a[i])
            k = pr[k-1];
        if(a[k] == a[i])
            k++;
        pr[i] = k;
    }

    k = 0;
    for(int i = 1 ; i < b.size() ; i++)
    {
        while(k != 0 && a[k] != b[i])
            k = pr[k-1];
        if(a[k] == b[i])
            k++;
        if(k >= a.size() - 1)
        {
            //cout << i - (a.size() - 1) + 1 << " ";
            ans.push_back(i - (a.size() - 1) + 1);
        }
    }
    fout << ans.size() << '\n';
    for(int i = 0 ; i < min(1000, (int)ans.size()) ; i++)
        fout << ans[i] << " ";
    fout << '\n';
    return 0;
}