KMP 模板

#include <cstdio>
#include <cstring>
#include <cctype>
#include <cmath>
#include <set>
#include <map>
#include <list>
#include <queue>
#include <deque>
#include <stack>
#include <string>
#include <bitset>
#include <vector>
#include <iostream>
#include <algorithm>
#include <stdlib.h>

using namespace std;
typedef long long LL;
const int INF=2e9+1e8;
const int MOD=1e9+7;
const int MAX_SIZE=1005;

void set_kmp_next(int *kmp_next,int len,string& b)
{
    int i=0,j=-1;
    kmp_next[0]=-1;
    while(i<len)
    {
        if(j==-1||b[i]==b[j])
        {
            i++;
            j++;
            kmp_next[i]=j;
        }
        else
            j=kmp_next[j];
    }
}
int kmp(string &a,string& b)  // a 为原串  ，b 为模式串
{
    int i,j,lena=a.size(),lenb=b.size();
    int kmp_next[MAX_SIZE];
    set_kmp_next(kmp_next,lenb,b);
    i=j=0;
    while(i<lena)
    {
        if(j==-1||a[i]==b[j])
        {
            i++;
            j++;
        }
        else
            j=kmp_next[j];
        if(j==lenb)
            return i-j+1;
    }
    return -1;
}
int main()
{
    string a,b;
    while(cin>>a>>b)
    {
        cout<<kmp(a,b)<<endl;
    }
    return 0;
}