hdu 1711

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/qq_40924940/article/details/83991234

题目大意,给两个长度分别为n m数组 a b

寻找一个k 使得 存在  a[k] == b[1],a[k+1] =b[2] ........ a[k+m-1] = b[m]

这就很简单了。。一个kmp泡一下,当 j 跑到结束的时候  i  ==  k + m - 1

直接输出就好 如果不行就返回 -1

以下为AC代码

#include<iostream>
#include<cstdio>
using namespace std;
const int maxn = 1e7+5;
int nex[maxn];
int c1[maxn],c2[maxn];
int l1,l2;
void getnex()
{
	int j=0,k=-1;
	nex[0]=-1;
	while(j<l2)
	{
		if(k==-1 || c2[k]==c2[j])
		nex[++j]=++k;
		else
		k=nex[k];
	}
}
int kmp()
{
	int i=0,j=0;
	getnex();
	while(i<l1)
	{
		if(j==-1 || c1[i]==c2[j])
		{
			i++,j++;
		}
		else
		{
			j = nex[j];
		}
		if(j == l2)
		{
			return i;
		}
	}
	return -1;
}
int main()
{
	int t;
	scanf("%d",&t);
	while(t--)
	{
		scanf("%d%d",&l1,&l2);
		for(int i=0;i<l1;i++)
		scanf("%d",&c1[i]);
		for(int i=0;i<l2;i++)
		scanf("%d",&c2[i]);
		if(kmp()!=-1)
		printf("%d\n",kmp()-l2+1);
		else
		printf("-1\n");
	}
	return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_40924940/article/details/83991234