http://acm.hdu.edu.cn/showproblem.php?pid=1711
看了 严蔚敏老师讲解的KMP算法后,想找个题做一下,听的时候,感觉还行,可是做起题来还是有点问题,特别是next数组的取值,自己参考着写一直没有过,
后来看了下别人的资料,next[0]从-1开始取值就AC了,看来next函数的取值还是比较重要,理解还是不够。。。。
#include<iostream>
#define M 10005
#define N 1000005
using namespace std;
int s[N],t[M],next[M];
int n,m;
void getNext()
{
int i=0;
int j=-1;
next[0]=-1;
while(i<m)
{
if(t[i]==t[j]||j==-1)
{
i++;
j++;
if(t[i]!=t[j])
next[i]=j;
else
next[i]=next[j];
}
else
j=next[j];
}
}
int KMP(int pos)
{
int i=pos;
int j=0;
while(i<n&&j<m)
{
if(s[i]==t[j]||j==-1)
{
i++;
j++;
}
else
j=next[j];
}
if(j==m)
return i-j+1;
else
return -1;
}
int main()
{
int i,T;
cin>>T;
while(T--)
{
memset(s,0,sizeof(s));
memset(t,0,sizeof(t));
memset(next,0,sizeof(next));
scanf("%d%d",&n,&m);
for(i=0;i<n;i++)
scanf("%d",&s[i]);
for(i=0;i<m;i++)
scanf("%d",&t[i]);
getNext();
printf("%d\n",KMP(0));
}
return 0;
}