Cod sursa(job #1327758)

Utilizator theprdvtheprdv theprdv Data 27 ianuarie 2015 05:52:24
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 0.95 kb
#include <fstream>
#include <string.h>
using namespace std;
fstream fin("strmatch.in", ios::in);
fstream fout("strmatch.out", ios::out);

#define MAX 2000010
char N[MAX], M[MAX];
int n, m, pi[MAX], d[1001], t;

void prefix()
{
    int q=0;
    pi[1]=0;
    for(int i=2; i<=n; i++){
        while(N[i]!=N[q+1] && q>0)
            q=pi[q];
        if(N[q+1]==N[i])
            q++;
        pi[i]=q;
    }
}
void KMP()
{
    int q=0;
    for(int i=1; i<=m; i++){
        while(N[q+1]!=M[i] && q>0)
            q=pi[q];
        if(M[i]==N[q+1])
            q++;
        if(q==n){
            t++;
            if(t<1000) d[t]=i-n;
        }
    }
}

int main()
{
    fin.getline(N+1, MAX);    ::n=strlen(N+1);
    fin.getline(M+1, MAX);    ::m=strlen(M+1);
    prefix();
    KMP();
    fout<<t<<"\n";
    for(int i=1; i<=t; i++)
        fout<<d[i]<<" ";
    fout<<"\n";

    fin.close();
    fout.close();
    return 0;
}