HDU 1711 Number Sequence
求母串第几位开始能和子串匹配,不能则输出-1
#include<iostream>
#include<algorithm>
#include<stdio.h>
using namespace std;
const int maxn = 1e6+20;
int nxt[maxn];
int m;//母
int n;//子
int s[maxn];//母
int t[maxn];//子
void getnxt()
{
int i=1;
int j=0;
nxt[0]=0;
while(i<n)
{
if(t[j]==t[i])
{
nxt[i++]=++j;
}
else if(j==0)
{
i++;
}
else
{
j=nxt[j-1];
}
}
}
int kmp()
{
int i=0;//母
int j=0;//子
while(i<m&&j<n)
{
if(s[i]==t[j])
{
i++;
j++;
}
else if(j==0)
{
i++;
}
else
{
j=nxt[j-1];
}
}
if(j==n) return i-n+1;
else return -1;
}
int main()
{
int T;
scanf("%d",&T);
while(T--)
{
scanf("%d %d",&m,&n);
for(int i=0;i<m;i++) scanf("%d",&s[i]);
for(int i=0;i<n;i++) scanf("%d",&t[i]);
getnxt();
printf("%d\n",kmp());
}
return 0;
}