模板題:
#include <iostream>
#include <stdio.h>
using namespace std;
int NEXT[10010];
int s[1000010],p[10010];
int n,m;
void GetNEXTval()
{
int pLen = m;
NEXT[0] = -1;
int k = -1;
int j = 0;
while (j < pLen - 1)
{
//p[k]表示前綴,p[j]表示後綴
if (k == -1 || p[j] == p[k])
{
++j;
++k;
//較之前NEXT數組求法,改動在下面4行
if (p[j] != p[k])
NEXT[j] = k; //之前只有這一行
else
//因爲不能出現p[j] = p[ NEXT[j ]],所以當出現時需要繼續遞歸,k = NEXT[k] = NEXT[NEXT[k]]
NEXT[j] = NEXT[k];
}
else
{
k = NEXT[k];
}
}
}
int KmpSearch()
{
int i = 0;
int j = 0;
int sLen = n;
int pLen = m;
while (i < sLen && j < pLen)
{
//①如果j = -1,或者當前字符匹配成功(即S[i] == P[j]),都令i++,j++
if (j == -1 || s[i] == p[j])
{
i++;
j++;
}
else
{
//②如果j != -1,且當前字符匹配失敗(即S[i] != P[j]),則令 i 不變,j = NEXT[j]
//NEXT[j]即爲j所對應的NEXT值
j = NEXT[j];
}
}
if (j == pLen)
return i - j;
else
return -1;
}
int main()
{
int t;cin>>t;
while(t--)
{
cin>>n>>m;
for(int i=0;i<n;i++)
scanf("%d",&s[i]);
for(int i=0;i<m;i++)
scanf("%d",&p[i]);
GetNEXTval();
int ans=KmpSearch();
if(ans==-1) cout<<-1<<endl;
else cout<<ans+1<<endl;
}
return 0;
}