Cod sursa(job #1635990)

Utilizator xtreme77Patrick Sava xtreme77 Data 6 martie 2016 21:28:12
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.91 kb
/**
 * Code by Patrick Sava
 * "Spiru Haret" National College of Bucharest
 **/

# include "fstream"
# include "cstring"
# include "vector"
# include "queue"
# include "bitset"
# include "algorithm"
# include "map"
# include "set"
# include "unordered_map"
# include "deque"
# include "string"
# include "iomanip"
# include "cmath"
# include "stack"
# include "cassert"

const char IN [ ] =  "strmatch.in" ;
const char OUT [ ] = "strmatch.out" ;

using namespace std ;

# define pb push_back
# define mp make_pair
# define FORN( a , b , c ) for ( register int a = b ; a <= c ; ++ a )
# define FORNBACK( a , b , c ) for ( register int a = b ; a >= c ; -- a )

ifstream cin ( IN ) ;
ofstream cout ( OUT ) ;

const int MAX = 4e6 + 14 ;

char sir [ MAX ] ;

int z [ MAX ] ;

int cate ;

vector < int > sol ;

int main()
{
    cin >> ( sir + 1 ) ;
    int l = strlen( sir + 1 ) ;
    sir [ l + 1 ] = '#' ;
    cin >> ( sir + l + 2 ) ;
    int n = strlen ( sir + 1 ) ;
    int L = 1 ;
    int R = 1 ;
    z [ 1 ] = 0 ;
    FORN ( i , 2 , n )
    {
        if ( i > R )
        {
            while ( i + z [ i ] <= n and sir [ i + z [ i ] ] == sir [ z [ i ] + 1 ] ) ++ z [ i ] ;
            L = i ;
            R = i + z [ i ] - 1 ;
        }
        else {
            int oo = i - L + 1 ;
            if ( i + z [ oo ] - 1 >= R ) {
                z [ i ] = R - i ;
                while ( i + z [ i ] <= n and sir [ i + z [ i ] ] == sir [ z [ i ] + 1 ] ) ++ z [ i ] ;
                L = i ;
                R = i + z [ i ] - 1 ;
            }
            else {
                z [ i ] = z [ oo ] ;
            }
        }

        if ( z [ i ] == l ) {
            ++ cate ;
            if ( cate <= 1000 ) {
                sol.pb ( i - l - 2 ) ;
            }
        }
    }
    cout << cate << '\n' ;
    for ( auto x : sol )
        cout << x << ' ' ;
    return 0;
}