Cod sursa(job #1379296)

Utilizator NacuCristianCristian Nacu NacuCristian Data 6 martie 2015 17:19:20
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.26 kb
#include <iostream>
#include <fstream>
#include <string.h>
#include <vector>

using namespace std;

char s1[2000100],s2[2000100];
int pi[2000100],m,n,nr;
int sol[1024];

void citire()
{
    freopen("strmatch.in", "r", stdin);
    fgets(s1, sizeof(s1), stdin);
    fgets(s2, sizeof(s2), stdin);
    s1[strlen(s1)-1]=s1[strlen(s1)];
    s2[strlen(s2)-1]=s2[strlen(s2)];
    m=strlen(s1),n=strlen(s2);
    for(int i=m;i>=0;i--)
        s1[i+1]=s1[i];
    s1[0]=' ';
    for(int i=n;i>=0;i--)
        s2[i+1]=s2[i];
    s2[0]=' ';
}

void prefix()
{
    int q=0;
    pi[1]=0;
    for(int i=2;i<=m;i++)
    {
        while(q && s1[i]!=s1[q+1])
            q=pi[q];
        if(s1[i]==s1[q+1])
            q++;
        pi[i]=q;
    }
}


void kmp()
{

    prefix();
    for(int i=1,k=0;i<=n;i++)
    {
        while(k && s1[k+1]!=s2[i])
            k=pi[k];
        if(s1[k+1]==s2[i])
            k++;
        if(k==m)
        {
            k=pi[m];
            ++nr;
            if(nr<=1000)
                sol[nr]=i-m;
        }
    }
}


int main()
{
    citire();
    kmp();
    freopen("strmatch.out","w",stdout);
    printf("%d\n",nr);
    for(int i=1;i<=min(nr,1000);i++)
        printf("%d ",sol[i]);
    return 0;
}