Pagini recente » Cod sursa (job #1956876) | Istoria paginii preoni-2007/runda-3/solutii | Cod sursa (job #177312) | Cod sursa (job #2533728) | Cod sursa (job #434592)
Cod sursa(job #434592)
/*
* 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;
}