KMP模板(HDU1711)

#include<stdio.h>
#include<math.h>
#include<string.h>
#include<stack>
#include<set>
#include<queue>
#include<vector>
#include<iostream>
#include<algorithm>
#define MAXN 1010100
#define LL long long
#define ll __int64
#define INF 0x7fffffff
#define cs(s) freopen(s,"r",stdin)
#define mem(x) memset(x,0,sizeof(x))
#define PI acos(-1)
#define eps 1e-10
using namespace std;
int gcd(int a,int b){return b?gcd(b,a%b):a;}
int lcm(int a,int b){return a/gcd(a,b)*b;}
LL powmod(LL a,LL b,LL MOD){LL ans=1;while(b){if(b%2)ans=ans*a%MOD;a=a*a%MOD;b/=2;}return ans;}
//head
int t,m,n;
int pattern[1101001],text[1001001];
vector <int> find_substring(int pattern[],int text[]){
	vector<int> nex (n+1,0);
	for(int i=1;i<n;i++){
		int j=i;
		while(j>0) {
		    j=nex[j];
		    if(pattern[j]==pattern[i]){nex[i+1]=j+1;break;}
		}
	}
	vector<int>pos;
	for(int i=0,j=0;i<m;i++){
		if(j<n&&text[i]==pattern[j])j++;
		else{
			while(j>0){
				j=nex[j];
				if(text[i]==pattern[j]){
					j++;
					break;
				}
			}
		}
		if(j==n)pos.push_back(i-n+1);
	}
	return pos;
}
int main(){
	ios::sync_with_stdio(false);
	for(cin>>t;t;t--){
		cin>>m>>n;
		for(int i=0;i<m;i++)cin>>text[i];
		for(int i=0;i<n;i++)cin>>pattern[i];
		vector<int>ans=find_substring(pattern,text);
		if(!ans.size())cout<<-1<<'\n';
		else cout<<ans[0]+1<<'\n';
	}
	return 0;
}
全部评论

相关推荐

不愿透露姓名的神秘牛友
07-24 13:35
点赞 评论 收藏
分享
每晚夜里独自颤抖:你cet6就cet6,cet4就cet4,你写个cet证书等是什么意思。专业技能快赶上项目行数,你做的这2个项目哪里能提现你有这么多技能呢
点赞 评论 收藏
分享
07-23 14:04
东北大学 C++
既然这样,为什么不点击就送呢
牛马88号:因为你合适。但有很多笔试就挂了、通过了再排序的
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务