Cod sursa(job #1226595)

Utilizator RaileanuCristian Raileanu Raileanu Data 6 septembrie 2014 13:09:18
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 0.8 kb
#include <fstream>
#include<string>

using namespace std;
#define MX 2000010
ifstream f1("strmatch.in");
ofstream f2("strmatch.out");
string a,b;
int pi[MX], k, i, n,m, ns, sol[MX] ;

int main()
{
    f1>>a>>b;

    n=a.length();
    m=b.length();

    a=' '+a;
    b=' '+b;

    k=0;
    pi[1]=0;
    for (i=2; i<=n; i++ )
    {   while (k>0 && a[k+1]!=a[i] )
            k=pi[k];
        if (a[k+1]==a[i] )
            k++;
        pi[i]=k; }

    k=0;
    for (i=1; i<=m ; i++)
        {   while (k>0 && a[k+1]!=b[i] )
                k=pi[k];
            if (a[k+1]==b[i] )
                k++;
            if (k==n)
                sol[++ns]=i-n;
        }

    f2<<ns<<"\n";
    for (i=1; i<=ns && i<=1000; i++)
        f2<<sol[i]<<" ";

    f2.close();
    return 0;
}