【codevs1204】寻找子串位置 kmp

原题

//codevs1204 寻找子串位置
#include<iostream>
#include<string>
#include<cstring>
#include<cstdio>
using namespace std;

string a,b;
int n,m;
int fail[101000];

void getfail()
{
	fail[0]=0;fail[1]=0;
	for (int i=1;i<m;i++)
	{
		int j=fail[i];
		while(j&&b[i]!=b[j]) j=fail[j];
		fail[i+1]=b[i]==b[j]?j+1:0;
	}
}

int main()
{
	cin>>a>>b;
	n=a.length();m=b.length();
	getfail();
	int j=0;
	for (int i=0;i<n;i++)
	{
		while(j&&a[i]!=b[j]) j=fail[j];
		if (a[i]==b[j]) j++;
		if (j==m) printf("%d\n",i-j+2); 
	}

	return 0;
}