Cod sursa(job #840445)

Utilizator VisuianMihaiMihai Visuian VisuianMihai Data 22 decembrie 2012 17:50:04
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 1.01 kb
#include <fstream>
#include <string>
using namespace std;
ifstream fin("strmatch.in");
ofstream fout("strmatch.out");
string a, b, s;
int l, r, z[4000004], nr, n, sol[2000001];
int main()
{
    fin >> a >> b;
    s = a+'$'+b;
    n = s.size();
    for ( int i = 1; i < n; i++ )
    {
        if ( i > r )
        {
            l = r = i;
            while ( r < n && s[r-l] == s[r] )
                r++;
            z[i] = r-l;
            r--;
        }
        else
        {
            int k = i-l;
            if ( z[k] < r-i+1 )
                z[i] = z[k];
            else
            {
                l = i;
                while ( r < n && s[r-l] == s[r] )
                    r++;
                z[i] = r-l;
                r--;
            }
        }
        if ( z[i] == a.size() )
            sol[++nr] = i - z[i] - 1;
    }
    fout << nr << '\n';
    for ( int i = 1; i <= nr; i++ )
        fout << sol[i] << ' ';
    fin.close();
    fout.close();
    return 0;
}