kmp算法笔记
一般匹配字符串时,我们从目标字符串str(假设长度为n)的第一个下标选取和ptr长度(长度为m)一样的子字符串进行比较,如果一样,就返回开始处的下标值,不一样,选取str下一个下标,同样选取长度为n的字符串进行比较,直到str的末尾(实际比较时,下标移动到n-m)。这样的时间复杂度是O(n*m)。
ababaca
这里我们要计算一个长度为m的转移函数next。
next数组的含义就是一个固定字符串的最长前缀和最长后缀相同的长度。
比如:abcjkdabc,那么这个数组的最长前缀和最长后缀相同必然是abc。
cbcbc,最长前缀和最长后缀相同是cbc。
abcbc,最长前缀和最长后缀相同是不存在的。
注意最长前缀:是说以第一个字符开始,但是不包含最后一个字符。
比如aaaa相同的最长前缀和最长后缀是aaa。
对于目标字符串ptr,ababaca,长度是7,所以next[0],next[1],next[2],next[3],next[4],next[5],next[6]分别计算的是
a,ab,aba,abab,ababa,ababac,ababaca的相同的最长前缀和最长后缀的长度。由于a,ab,aba,abab,ababa,ababac,ababaca的相同的最长前缀和最长后缀是“”,“”,“a”,“ab”,“aba”,“”,“a”,所以next数组的值是[-1,-1,0,1,2,-1,0],这里-1表示不存在,0表示存在长度为1,2表示存在长度为3。这是为了和代码相对应。
KMP算法:可以实现复杂度为O(m+n)
ac代码:kmp模板体
#include<cstdio>
#include<iostream>
#include<algorithm>
#include<cstring>
using namespace std;
int a[1000000];
int b[10000];
int next1[10000];
void get_next(int m)
{
int k=-1;
next1[0]=-1;
for(int i=1;i<m;i++)
{
while(k>-1 && b[k+1]!=b[i])
k=next1[k];
if(b[k+1]==b[i])
k=k+1;
next1[i]=k;
}
}
int fang(int n,int m)
{
get_next(m);
int k=-1;
for(int i=0;i<n;i++)
{
while(k>-1 && b[k+1]!=a[i])
{
k=next1[k];
}
if(a[i]==b[k+1])
k=k+1;
if(k==m-1)
return i-m+2;
}
return -1;
}
int main()
{
int t;int n,m;
cin>>t;
while(t--)
{
cin>>n>>m;
for(int i=0;i<n;i++)
{
cin>>a[i];
}
for(int i=0;i<m;i++)
{
cin>>b[i];
}
int ans=fang(n,m);
cout<<ans<<endl;
}
}