Cod sursa(job #2791191)

Utilizator Robert.BrindeaBrindea Robert Robert.Brindea Data 30 octombrie 2021 10:47:18
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.97 kb
#include <iostream>
#include <fstream>
#include <vector>

using namespace std;

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

const int MAXN = 2000005;
string a, b;
int p[MAXN], n, m, rez;
vector<int> sol;

void generare_prefixe()
{
    int i = 0, j = p[0] = -1;
    while(i < n)
    {
        while(j>=0 && a[i] != a[j])
            j = p[j];
        p[++i] = ++j;
    }
}

void kmp()
{
    int i = 0, j = 0;
    while(i < m)
    {
        while(j>=0 && b[i] != a[j])
            j = p[j];
        i++;
        j++;
        if(j == n)
        {
            j = p[j];
            rez++;
            if(rez <= 1000)
                sol.push_back(i-n);
        }
    }
}

int main()
{
    fin >> a >> b;
    n = a.size(); m = b.size();
    generare_prefixe();
    kmp();
    fout << rez << "\n";
    int len = min((int)sol.size(), 1001);
    for(int i = 0; i < len; i++)
        fout << sol[i] << " ";
    return 0;
}