题目链接:点击打开链接
题目大意:给你俩数列,一个有n个数字,一个有m个数字,问你这n个数字里面是否包含连着的那m个数字,如果有的话是在哪一个位置开始的,如果没有的话就输出-1
题目思路:就是KMP的纯模板题。首先先对第二个数列即m个数字的那个数列进行预处理来获取Next数组(这里有个小坑,如果定义小写的next的数组,可能会跟某些头文件产生冲突从而导致CE),就是定义一个j=0,k=-1,如果k==-1或者b[j]和b[k]一样的话,就直接让j++,k++,并且让当前的next[j]=k,如果遇到不相等,那就开始递归找,k=next[k],一直找到跟当前位置能够产生前后缀相同的最大数字。然后就是KMP主算法,一个j=0,k=0,j作为主串中的指针,k作为模式串中的指针。然后while循环判断条件是j<m。然后如果k==-1或者b[j]和b[k]一样的话,就直接让j++,k++就可以了,否则跟上面一样递归,没错就第一个判断条件比生成next数组少了一个next[j]=k。最后看看模式串有没有被匹配完,如果能匹配完,就说明能够在主串中找到,输出i-m+1就是在主串中出现的数字,如果没有匹配完,就说明找不到,输出-1。
以下是代码:
#include <iostream>
#include<cstdio>
#include<cstring>
#include<map>
#include<string>
#include<iomanip>
using namespace std;
int a[1000005],b[10005],Next[10005];
int main()
{
int t,n,m;
scanf("%d",&t);
while(t--){
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++){
scanf("%d",&a[i]);
}
for(int i=0;i<m;i++){
scanf("%d",&b[i]);
}
int j=0,k=-1;
Next[0]=-1;
while(j<m){
if(k==-1||b[j]==b[k]){
j++,k++;
Next[j]=k;
}
else{
k=Next[k];
}
}
j=0,k=0;
while(j<n){
if(k==-1||a[j]==b[k]){
j++,k++;
}
else{
k=Next[k];
}
if(k==m){
break;
}
}
if(k==m){
printf("%d\n",j-m+1);
}
else{
printf("-1\n");
}
}
return 0;
}