Cod sursa(job #1092715)

Utilizator fhandreiAndrei Hareza fhandrei Data 27 ianuarie 2014 12:47:46
Problema Potrivirea sirurilor Scor 100
Compilator cpp Status done
Runda Arhiva educationala Marime 1.29 kb
// Include
#include <fstream>
#include <cstring>
using namespace std;

// Constante
const int sz = (int)2e6+1;

// Functii
void makePrefix();

// Variabile
ifstream in("strmatch.in");
ofstream out("strmatch.out");

char str[sz], sub[sz];
int strLen, subLen;

int prefix[sz];

int answer;
int answers[sz];

// Main
int main()
{
	in >> (sub+1) >> (str+1);
	
	subLen = strlen(sub+1);
	strLen = strlen(str+1);
	
	makePrefix();
	
	int currentPrefix = 0;
	for(int i=1 ; i<=strLen ; ++i)
	{
		while(currentPrefix && sub[currentPrefix+1] != str[i])
			currentPrefix = prefix[currentPrefix];
		
		if(sub[currentPrefix+1] == str[i])
			++currentPrefix;
		
		if(currentPrefix == subLen)
		{
			if(++answer <= 1000)
				answers[answer] = i - subLen;
			
			currentPrefix = prefix[currentPrefix];
		}
	}
	
	out << answer << '\n';
	int limit = min(1000, answer);
	for(int i=1 ; i<=limit ; ++i)
		out << answers[i] << ' ';
	
	out << '\n';
	
	in.close();
	out.close();
	return 0;
}

void makePrefix()
{
	int currentPrefix = 0;
	
	for(int i=2 ; i<=subLen ; ++i)
	{
		while(currentPrefix && sub[currentPrefix+1] != sub[i])
			currentPrefix = prefix[currentPrefix];
		
		if(sub[currentPrefix+1] == sub[i])
			++currentPrefix;
		
		prefix[i] = currentPrefix;
	}
}