Cod sursa(job #1327756)

Utilizator theprdvtheprdv theprdv Data 27 ianuarie 2015 05:42:08
Problema Potrivirea sirurilor Scor 16
Compilator cpp Status done
Runda Arhiva educationala Marime 0.92 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++;
        d[i]=q;
        if(q==n) t++;
    }
}

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; m>1000 ? i<=1000 : i<m; i++)
        if(d[i]==n) fout<<i-n<<" ";

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