Cod sursa(job #1486697)

Utilizator afkidStancioiu Nicu Razvan afkid Data 15 septembrie 2015 13:52:20
Problema Potrivirea sirurilor Scor 40
Compilator cpp Status done
Runda Arhiva educationala Marime 0.95 kb
#include <bits/stdc++.h>
#define M 2000100

using namespace std;
inline int MAX(int a,int b){return (a>b)?a:b;}
inline int MIN(int a,int b){return (a<b)?a:b;}
int n,m;
char s[M],p[M];
vector<int> match;
int kmpNext[M];

void preKmp()
{
	int k;
	kmpNext[1]=0;
	for(int i=2;i<=n;++i)
	{
		while(k>0 && p[i]!=p[k+1])
			k=kmpNext[k];
		if(p[k+1]==p[i])
			k++;
		kmpNext[i]=k;
	}
}

void KMP()
{
	int cnt=0;
	preKmp();
	int k=0;
	for(int i=1;i<=m;++i)
	{
		while(k>0 && p[k+1]!=s[i])
			k=kmpNext[k];
		if(p[k+1]==s[i])
			k++;
		if(k==n)
		{
			++cnt;
			if(cnt<=1000)
				match.push_back(i-n);
		}
	}
}

int main()
{
	freopen("strmatch.in","r",stdin);
	freopen("strmatch.out","w",stdout);
	ios_base::sync_with_stdio(0);
	//cin.tie(0);
	cin>>p+1;
	cin>>s+1;
	m=strlen(s+1);
	n=strlen(p+1);
	KMP();
	cout<<match.size()<<"\n";
	for(int i=0;i<match.size();++i)
	{
		cout<<match[i]<<" ";
	}
	cout<<"\n";
	return 0;
}