Cod sursa(job #2682103)

Utilizator andrei.florea0405Florea Andrei-Bogdan andrei.florea0405 Data 7 decembrie 2020 19:42:43
Problema Potrivirea sirurilor Scor 100
Compilator cpp-64 Status done
Runda Arhiva educationala Marime 1 kb
#include <bits/stdc++.h>
using namespace std;
#define NMax 2000005
	
int M, N;
char A[NMax], B[NMax];
int pi[NMax], pos[1024];

	
void make_prefix(void)	
{
	int i, q = 0;
    pi[1] = 0;
	for (i = 2, pi[1] = 0; i <= M; ++i) {
		while (q && A[q+1] != A[i])
			q = pi[q];
	
		if (A[q+1] == A[i])
			++q;
	
		pi[i] = q;
	}
}
	
 
	
int main() {
	int i, q = 0, count = 0;
	
    ifstream fin("strmatch.in");
    ofstream fout("strmatch.out");
	
    fin.getline(A, NMax);
    fin.getline(B, NMax);
	
    M = strlen(A);
    N = strlen(B);
	
	for (i = M; i; i--) 
        A[i] = A[i-1]; 
    A[0] = ' ';
	
    for (i = N; i; i--) 
        B[i] = B[i-1]; 
    B[0] = ' ';	
	
	make_prefix();
	
	for (i = 1; i <= N; i++) {		
		while (q && A[q+1] != B[i])
			q = pi[q];
	
		if (A[q+1] == B[i])
			++q;
	
		if (q == M) {
			q = pi[M];
			count++;
			if (count <= 1000)
				pos[count] = i - M;
		}
	}
	
	fout << count << "\n";
	
	for (i = 1; i <= min(count, 1000); i++)
        fout << pos[i] << " ";
	
	fout << "\n";
	
			
	
	return 0;
	
}