Cod sursa(job #433987)

Utilizator alexandru92alexandru alexandru92 Data 4 aprilie 2010 21:07:30
Problema Potrivirea sirurilor Scor 14
Compilator cpp Status done
Runda Arhiva educationala Marime 1.56 kb
/* 
 * File:   main.cpp
 * Author: SpeeDemon
 *
 * Created on April 4, 2010, 6:33 PM
 */
#include <string>
#include <vector>
#include <cstdlib>
#include <fstream>
#include <iterator>
#define Alpha 131
#define Modulo 1000000007

/*
 *
 */
using namespace std;
vector< int > v;
string pattern, ss;
inline int min( int x, int y )
{
    if( x > y )
        return y;
    return x;
}
void Rabin_Karp( int PatternSize, int StringSize )
{
    if( PatternSize > StringSize )
        return;
    bool ok=true;
    int E, i, j, k, hp, hss, till=StringSize-PatternSize+1;
    for( E=1, hp=hss=0, i=PatternSize-1; i >= 0; --i )
    {
        hp+=( 1LL*pattern[i]*E )%Modulo;
        hss+=( 1LL*ss[i]*E )%Modulo;
        if( i > 0 )
            E=(1LL*E*Alpha)%Modulo;
        if( pattern[i] != ss[i] )
            ok=false;
    }
    if( ok )
        v.push_back(0);
    for( i=PatternSize; i < StringSize; ++i )
    {
        hss=( ( ( ( hss%Modulo - (1LL*ss[i-PatternSize]*E)%Modulo )%Modulo + Modulo )%Modulo )*(Alpha%Modulo) + ss[i]%Modulo  )%Modulo;
        if( hss == hp )
        {
            for( j=PatternSize-1, k=i; j >= 0; --j, --k )
                if( pattern[j] != ss[k] )
                    break;
            if( j < 0 )
                v.push_back(k+1);
        }
    }
}
int main( void )
{
    ifstream in( "strmatch.in" );
    in>>pattern>>ss;
    Rabin_Karp( pattern.size(), ss.size() );
    ofstream out( "strmatch.out" );
    out<<v.size()<<'\n';
    copy( v.begin(), v.begin()+min( v.size(), 1000 ), ostream_iterator<int>( out, " " ) );
    out<<'\n';
    return EXIT_SUCCESS;
}