Cod sursa(job #2606975)

Utilizator Cezar211Popoveniuc Cezar Cezar211 Data 28 aprilie 2020 22:59:42
Problema Potrivirea sirurilor Scor 60
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.26 kb
#include <bits/stdc++.h>
#define NM 2000005
///KMP
using namespace std;
ifstream fin ("strmatch.in");
ofstream fout ("strmatch.out");
char a[NM], c;
int rez, m, kmp[NM];
void build(char a[], int kmp[])
{
    int i = 0, j = 1;
    kmp[0] = 0;
    for(j=1; j<m; j++)
        if(a[i] == a[j])
        {
            kmp[j] = i+1;
            i++;
        }
        else
        {
            while(i!=0 && a[i]!=a[j])
                i = kmp[i-1];
            if(a[i] == a[j])
                kmp[j] = i+1;
            else
                kmp[j] = i;
        }
}
vector<int> sol;
int main()
{
    fin.getline(a, NM);
    m = strlen(a);
    build(a, kmp);
    int i = 0;
    for(int j=0; fin >> c; j++)
    {
        if(c == a[i])
        {
            if(i == m-1)
            {
                if(rez <= 1000)
                    sol.push_back(j-m+1);
                rez++;
                i = kmp[i];
            }
            else
                i++;
        }
        else
        {
            while(i!=0 && c!=a[i])
                i = kmp[i-1];
            if(c == a[i])
                i++;
        }
    }
    fout << rez << '\n';
    for(int i=0; i<min(rez, 1000); i++)
        fout << sol[i] << ' ';
    return 0;
}