Cod sursa(job #463917)

Utilizator darrenRares Buhai darren Data 17 iunie 2010 22:40:07
Problema Potrivirea sirurilor Scor 16
Compilator cpp Status done
Runda Arhiva educationala Marime 0.84 kb
#include<cstring>
#include<fstream>
#include<vector>
using namespace std;

char a[2000001], b[2000001];
int prefix[2000001];
vector<int> pos;

void make_prefix()
{
	int k = -1;
	prefix[0] = -1;
	for (int i = 1; i < strlen(a); ++i)
	{
		while (k != -1 && a[k + 1] != a[i])
			k = prefix[k];
		if (a[k + 1] == a[i])
			++k;
		prefix[i] = k;
	}
}

int main()
{
	ifstream fin("strmatch.in");
	ofstream fout("strmatch.out");
	fin >> a >> b;
	
	make_prefix();
	
	
	int ok = -1;
	for (int i = 0; i < strlen(b); ++i)
	{
		if (a[ok + 1] == b[i])
			++ok;
		else
			while (ok != -1 && a[ok + 1] != b[i])
				ok = prefix[ok];
		
		if (ok == strlen(a) - 1)
		{
			pos.push_back(i - strlen(a) + 1);
			ok = prefix[ok];
		}
	}
	fout << pos.size() << '\n';
	for (int i = 0; i < pos.size(); ++i)
		fout << pos[i] << ' ';
}