hdu 1711

最新推荐文章于 2019-10-01 15:18:00 发布

LXY_XYL

最新推荐文章于 2019-10-01 15:18:00 发布

阅读量72

点赞数

分类专栏： ACM c++

本文链接：https://blog.csdn.net/qq_40924940/article/details/83991234

版权

ACM c++ 专栏收录该内容

65 篇文章 0 订阅

订阅专栏

题目大意，给两个长度分别为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;
}