HDU-1711 Number Sequence (kmp)

题目：

http://acm.hdu.edu.cn/showproblem.php?pid=1711

题意：

在a串中找b串的位置，没有就返回-1。

思路：

kmp模版题，构造next[]扫一遍a就行了。

代码：

#include <bits/stdc++.h>
#define PI acos(-1.0)
#define INF 0x3f3f3f3f
#define MOD 1000000007
#define EPS 1e-6
#define N 1123456
using namespace std;
int n,m,sum,res,flag;
int a[N],b[N];
void getNext(int *pre,int len, int *next)
{
    int i=0,j=-1;
    next[0]=-1;
    while(i<len)
    {
        if(j==-1 || pre[i]==pre[j])
            i++,j++,next[i]=j;
        else
            j=next[j];
    }
}
int kmp(int *pre,int *str)
{
    int lenp=m,lens=n;
    int next[lenp];
    getNext(pre,lenp,next);
    int i,j;
    i=j=0;
    while(i<lens&&j<lenp)
    {
        if(j==-1||str[i]==pre[j])
            i++,j++;
        else
            j=next[j];
    }
    if(j>=lenp)return i-lenp;
    else return -2;
}
int main()
{
    int i,j,k,kk,cas,T,t,x,y,z;
    scanf("%d",&T);
    while(T--)
    {
        scanf("%d%d",&n,&m);
        for(i=0;i<n;i++)
            scanf("%d",&a[i]);
        for(i=0;i<m;i++)
            scanf("%d",&b[i]);
        printf("%d\n",kmp(b,a)+1);
    }
    return 0;
}