Cod sursa(job #434592)

Utilizator alexandru92alexandru alexandru92 Data 6 aprilie 2010 08:47:29
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.67 kb
/* 
 * File:   main.cpp
 * Author: VirtualDemon
 *
 * Created on April 5, 2010, 7:57 PM
 */
#include <string>
#include <vector>
#include <cstdlib>
#include <fstream>
#include <iterator>

/*
 * 
 */
using namespace std;
vector< int > v, f;
string pattern, ss;
inline void build_function( int PatternSize )
{
    int i, j;
    f.resize( PatternSize+1 );
    for( i=2; i <= PatternSize; ++i )
    {
        for( j=f[i-1]; ; j=f[j] )
        {
            if( pattern[j] == pattern[i-1] )
            {
                f[i]=j+1;
                break;
            }
            if( !j )
            {
                f[i]=0;
                break;
            }
        }
    }
}
inline void KMP( int PatternSize, int StringSize )
{
    if( PatternSize > StringSize )
        return;
    int i, j;
    build_function( PatternSize );
    for( i=j=0; j < StringSize; )
    {
        if( ss[j] == pattern[i] )
        {
            ++i, ++j;
            if( PatternSize == i )
                v.push_back(j-PatternSize);
        }
        else if( i > 0 )
                i=f[i];
             else ++j;
    }
}
inline int min( int x, int y )
{
    if( x > y )
        return y;
    return x;
}
int main(int argc, char** argv)
{
    ifstream in( "strmatch.in" );
    getline( in, pattern );
    getline( in, ss );
    char c=pattern[pattern.size()-1];
    if( ( c >= 'a' && c <= 'z' ) || ( c >= 'A' && c <= 'Z' ) || ( c >= '0' && c <= '9' ) )
        KMP( pattern.size(), ss.size() );
    else KMP( pattern.size()-1, 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;
}