Cod sursa(job #2259222)

Utilizator al.mocanuAlexandru Mocanu al.mocanu Data 13 octombrie 2018 10:36:36
Problema Potrivirea sirurilor Scor 0
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 0.63 kb
#include <fstream>
#include <cstring>
#define MAX 2000005
#define THRESH 1000
using namespace std;

char a[MAX], b[MAX];
int n, m, k, cnt, pi[MAX];

int main() {
	ifstream f("strmatch.in");
	ofstream g("strmatch.out");

	f >> a + 1 >> b + 1;
	n = strlen(a + 1);
	m = strlen(b + 1);

	k = 0;
	for (int i = 2; i <= n; ++i) {
		while (k && a[i] != a[k + 1])
			k = pi[k];
		if (a[k + 1] == a[i])
			++k;
		pi[i] = k;
	}

	k = 0;
	for (int i = 1; i <= m; ++i) {
		while (k && b[i] != a[k + 1])
			k = pi[k];
		if (b[i] == a[k + 1])
			++k;
		if (k == n) {
			++cnt;
			g << i - k << " ";
			if (cnt == THRESH)
				break;
		}
	}
	g << "\n";

	return 0;
}