Cod sursa(job #1146262)

Utilizator laurageorgescuLaura Georgescu laurageorgescu Data 18 martie 2014 20:34:51
Problema Potrivirea sirurilor Scor 72
Compilator cpp Status done
Runda Arhiva educationala Marime 1.62 kb
#include<fstream>
#include<vector>
#include<string>

using namespace std;

ifstream fin( "strmatch.in" );
ofstream fout( "strmatch.out" );

const int mod1 = 12974;
const int mod2 = 96235;

string a, b;
int v[ 1000 ];

inline int code( char c ) {
    if ( 'a' <= c && c <= 'z' ) {
        return c - 'A' + 26;
    } else {
        return c - 'A';
    }
}
int main()
{
    int p1, p2, ha1, ha2, hb1, hb2, sol, k;
    fin>>a>>b;
    sol = ha1 = ha2 = hb1 = hb2 = 0;
    p1 = p2 = 1;
    k = (int)a.size();

    if ( k < (int)b.size() ) {
        for( int i = k - 1; i >= 0; -- i ) {
            ha1 += p1*code( a[i] ); ha2 += p2*code( a[i] );
            hb1 += p1*code( b[i] ); hb2 += p2*code( b[i] );
            ha1 %= mod1; hb1 %= mod1;
            ha2 %= mod2; hb2 %= mod2;
            if ( i > 0 ) {
                p1 *= 52; p2 *= 52;
                p1 %= mod1; p2 %= mod2;
            }
        }
        if ( ha1 == hb1 && ha2 == hb2 ) {
            v[ 0 ] = 0;
            ++ sol;
        }
    }
    for( int i = k; i < (int)b.size(); ++ i ) {
        hb1 = ( hb1 - p1 * code( b[i-k] ) ) * 52 + code( b[i] );
        hb2 = ( hb2 - p2 * code( b[i-k] ) ) * 52 + code( b[i] );
        hb1 %= mod1; hb2 %= mod2;
        if ( hb1 < 0 ) {
            hb1 += mod1;
        }
        if ( hb2 < 0 ) {
            hb2 += mod2;
        }
        if ( ha1 == hb1 && ha2 == hb2 ) {
            if ( sol < 1000 )
                v[ sol ] = i - k + 1;
            ++ sol;
        }
    }

    fout<<sol<<'\n';

    sol = sol<1000?sol:1000;
    for( int i = 0; i < sol; ++ i ) {
        fout<<v[i]<<' ';
    }

    fin.close();
    fout.close();
    return 0;
}