Cod sursa(job #2722015)

Utilizator adiaioanaAdia R. adiaioana Data 12 martie 2021 15:50:35
Problema Potrivirea sirurilor Scor 38
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.1 kb
#include <fstream>
#include <cstring>
#include <vector>
#define NMAX 2000100
using namespace std;

ifstream cin("strmatch.in");
ofstream cout("strmatch.out");

char A[NMAX],B[NMAX], aux[NMAX];
int nA,nB, pre[NMAX];
vector <int> sol;
inline void scan() {

    A[0]=' ';
    B[0]=' ';
    cin>>aux;
    nA=strlen(aux);
    strcpy(A+1,aux);
    aux[0]='\0';
    cin>>aux;
    nB=strlen(aux);
    strcpy(B+1,aux);
}

inline void make_pre() {
    int q=0;
    for(int i=2; i<=nA; ++i)
    {
        while(q && A[q+1]!=A[i])
            q=pre[q];
        if(A[q+1]==A[i])
            q++;
        pre[i]=q;
    }
}

int main()
{
    scan();
    if(nA>nB)
    {
        cout<<0<<'\n';
        return 0;
    }
    make_pre();

    int q=0;
    for(int i=1; i<nB; ++i)
    {
        while(q && A[q+1]!=B[i])
            q=pre[q];
        if(A[q+1]==B[i])
            q++;
        if(q==nA)
            sol.push_back(i-nA);
    }

    cout<<sol.size()<<'\n';
    for(int i=0; i<min(1000, (int) sol.size()); ++i)
        cout<<sol[i]<<' ';
    cout<<'\n';
    return 0;
}