【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;
}