一道经典编程题目 Number Sequence 应用kmp算法的解决:
题目大意:给出两个数字序列(字符串也通用),在序列a中匹配到和序列b中完全相同的字串,返回第一次发生匹配的位置,
(如果有多次匹配,可以将结果存入vector,最后输出)。
样例输入:
2
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 1 3
13 5
1 2 1 2 3 1 2 3 1 3 2 1 2
1 2 3 2 1
样例输出:
6
-1
#include<iostream>
using namespace std;
const int MAXM=10000;
const int MAXN=1000000;
int nextTable[MAXM];
int pattern[MAXM];
int text[MAXN];
void getNextTable(int m){//创建next表
int j=0;
nextTable[j]=-1;
int i=nextTable[j];
while(j<m){
if(i==-1||pattern[j]==pattern[i]){
i++;j++;
if(j==m)return;
nextTable[j]=i;
}else i=nextTable[i];
}
return ;
}
int KMP(int n,int m){ //kmp算法找出第一个发生匹配的位置
getNextTable(m);
int i=0;
int j=0;
while(i<n&&j<m){
if(j==-1||text[i]==pattern[j]){
j++;
i++;
}else {
j=nextTable[j];
}
}
if(j==m){
return i-j+1;
}
else return -1;
}
int main(){
int caseNumber;
scanf("%d",&caseNumber);
while(caseNumber--){
int n;int m;
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++){
scanf("%d",&text[i]);
}
for(int i=0;i<m;i++){
scanf("%d",&pattern[i]);
}
printf("%d\n",KMP(n,m));
}
return 0;
}