Cod sursa(job #2099373)

Utilizator albucristianAlbu Cristian-Gabriel albucristian Data 4 ianuarie 2018 12:54:48
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.18 kb
#include <fstream>
#include <cstring>
using namespace std;
int pi[2000002],d[2000002];
char a[2000002],b[2000002];
int main()
{
    int n,m,i,c,k;
    ifstream in("strmatch.in");
    ofstream out("strmatch.out");
    in>>a+1>>b+1;
    m=strlen(b+1);
    n=strlen(a+1);
    k=0;
    pi[1]=0;
    for(i=2;i<=n;i++)
    {
        while(k>0&&a[i]!=a[k+1])
        {
            k=pi[k];
        }
        if(a[k+1]==a[i])
        {
            k++;
        }
        pi[i]=k;
    }
    k=0;
    c=0;
    for(i=1;i<=m;i++)
    {
        while(k>0&&b[i]!=a[k+1])
        {
            k=pi[k];
        }
        if(b[i]==a[k+1])
        {
            k++;
        }
        d[i]=k;
        if(k==n)
            c++;
    }
    if(c<=1000)
    {
        out<<c<<endl;
        for(i=1;i<=m;i++)
        {
            if(d[i]==n)
                out<<i-n<<" ";
        }
    }
    else
    {
        out<<c<<endl;
        c=1000;
        for(i=1;i<=m;i++)
        {
            if(d[i]==n)
            {
                out<<i-n<<" ";
                c--;
            }
            if(c==0)
                break;
        }
    }
    return 0;
}