Cod sursa(job #2374531)

Utilizator ZheroAlexandru Culea Zhero Data 7 martie 2019 19:11:04
Problema Potrivirea sirurilor Scor 8
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.01 kb
#include <iostream>
#include <fstream>
#include <vector>
using namespace std;

#define minn(a,b) (a<b? a : b)
const int LMAX=2e6 + 1;

string A,B;
int pi[LMAX];
void prefix()
{
    //indexarea incepe de la 0 deci unele valori din alg sunt decrementate cu 1
    pi[0]=0;
    int i=-1,y=A.size();
    for(int j=1;j<y;j++)
    {
        while(i>-1 && A[i+1]!=A[j])
            i=pi[i];
        if(A[i+1]==A[j])
            i++;
        pi[j]=i;
    }
}
int main()
{
    ifstream f("strmatch.in");
    ofstream g("strmatch.out");
    f>>A>>B;
    f.close();
    vector <int> sol;
    prefix();
    int i=-1,x,y;
    x=B.size();
    y=A.size();
    for(int j=0;j<x;j++)
    {
        while(i>-1 && A[i+1]!=B[j])
            i=pi[i];
        if(A[i+1]==B[j])
            i++;
        if(i==y-1)
        {
            sol.push_back(j-i);
            i=pi[i];
        }
    }
    g<<sol.size()<<'\n';
    x=minn(sol.size(),1000);
    for(i=0;i<x;i++)
        g<<sol[i]<<" ";
    return 0;
}