Cod sursa(job #2266817)

Utilizator cristibogdanPatrascu Cristian cristibogdan Data 22 octombrie 2018 21:43:27
Problema Potrivirea sirurilor Scor 80
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1.03 kb
#include <iostream>
#include <fstream>
#include <string.h>
#define NMAX 2000666
using namespace std;
ifstream f("strmatch.in");
ofstream g("strmatch.out");
int l1, l2, i, z[NMAX], sol, v[NMAX];
char s1[2 * NMAX], s2[2 * NMAX];

void zf(char p[])
{
    int l, r;
    int n = l1 + l2 + 1;
    z[0] = 0;
    l = 0;
    r = 0;
    for(i = 1; i < n; i++){

        if ( i <= r )
            z[i] = min(z[i - l], r - i );
        while( i + z[i] - 1 < n && p[ i + z[i] ] == p[ z[i] ])
            z[i] ++;
        if(z[i] == l1){
            sol ++;
            if(sol <= 1000)
                v[sol] = i - l1 - 1;
        }

        if(i + z[i] - 1 >= r)
        {
            r = i + z[i] - 1;
            l = i;
        }
    }
}
int main()
{
    f.getline(s1, NMAX);
    f.getline(s2, NMAX);
    l1 = strlen( s1 );
    l2 = strlen( s2 );
    s1[l1] = '#';
    strcat(s1 , s2 );
    zf(s1);
    g << sol << '\n';
    sol = min(sol, 1000);
    for(i = 1; i <= sol; i++)
        g << v[i] << " ";
    return 0;
}