Cod sursa(job #2198233)

Utilizator RaduVFVintila Radu-Florian RaduVF Data 23 aprilie 2018 22:37:31
Problema Potrivirea sirurilor Scor 38
Compilator cpp Status done
Runda Arhiva educationala Marime 0.82 kb
#include <bits/stdc++.h>
#define NMAX 2000005
using namespace std;

ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
int urm[NMAX];
char A[NMAX], B[NMAX];
int n, m, c;
vector <int> sol(NMAX);

void Next(char *A) {
    int k=-1;
    urm[0]=0;
    for(int i=1; i<=n; i++) {
        while(k>0 && A[k+1]!=A[i]) k=urm[k];
        if(A[k+1]==A[i]) k++;
        urm[i]=k;
    }
}

int main()
{
    int x=-1;
    fin.getline(A, NMAX); fin.get();
    fin.getline(B, NMAX);
    n=strlen(A); m=strlen(B);
    Next(A);
    for(int i=0; i<m; i++) {
        while(x>0 && A[x+1]!=B[i]) x=urm[x];
        if(A[x+1]==B[i]) x++;
        if(x==n-1) {
            sol[c++]=i-n+2;
            x=urm[x];
        }
    }
    fout<<c<<endl;
    for(int i=0; i<c; i++)
        fout<<sol[i]<<' ';
    return 0;
}