题意:
给出两个序列的数字,要你找到第一个序列中找到第二个序列并输出位置。
题解:
KMP模板题。
#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
const int MAXN=1000000+7;
int s[MAXN];
int p[MAXN];
int Next[MAXN];
int n,m;
void getNext()
{
Next[0]=-1;
int j=0,k=-1;
while(j<m)
{
if(k==-1||p[j]==p[k])
{
j++;
k++;
Next[j]=k;
}
else
k=Next[k];
}
}
void KMP()
{
getNext();
int k=0,j=0;
while(j<n&&k<m)
{
if(k==-1||s[j]==p[k])
{
j++;
k++;
}
else
k=Next[k];
if(k==m)
{
printf("%d\n",j-k+1);
return ;
}
}
printf("-1\n");
}
int main()
{
int t;
scanf("%d",&t);
while(t--)
{
scanf("%d%d",&n,&m);
for(int i=0;i<n;i++)
scanf("%d",&s[i]);
for(int i=0;i<m;i++)
scanf("%d",&p[i]);
KMP();
}
}