Cod sursa(job #494771)

Utilizator BitOneSAlexandru BitOne Data 22 octombrie 2010 21:14:48
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.09 kb
#include <string>
#include <vector>
#include <fstream>
#include <cstdlib>
#include <iterator>
#include <algorithm>
#define MAX_N 2000011

using namespace std;

/*
 *
 */
int f[MAX_N];
vector< int > r;
string pattern, s;
inline int min( int x, int y )
{
	if( x <= y )
		return x;
	return y;
}
inline void F( void )
{
	int M=pattern.size()-1, i, j;
	for( i=2; i <= M; ++i )
	{
		for( j=f[i-1]; j && pattern[i-1] != pattern[j]; j=f[j] );
		if( pattern[i-1] == pattern[j] )
			++j;
		f[i]=j;
	}
}
inline void KMP( void )
{
	int i, j, N=s.size()-1, M=pattern.size()-1;
	F();
	for( i=j=0; i < N; )
	{
		if( pattern[j] == s[i] )
		{
			++i, ++j;
			if( M == j )
				r.push_back(i-M);
		}
		else if( j )
				j=f[j];
			 else ++i;
	}
}
int main( void )
{
	ifstream in( "strmatch.in" );
	getline( in, pattern );
	getline( in, s );
	s.push_back(' ');
	pattern.push_back(' ');
	KMP();
	ofstream out( "strmatch.out" );
	out<<r.size()<<'\n';
	copy( r.begin(), r.begin()+min( 1000, r.size() ), ostream_iterator<int>( out, " " ) );
	out<<'\n';
	return EXIT_SUCCESS;
}